diff --git a/tests/test_musicbrainz.py b/tests/test_musicbrainz.py
index c672720..4dd956d 100644
--- a/tests/test_musicbrainz.py
+++ b/tests/test_musicbrainz.py
@@ -4,7 +4,6 @@ import importlib.util
 import json
 import sys
 import time
-from io import BytesIO
 from unittest.mock import MagicMock, patch
 
 # -- Load module directly ----------------------------------------------------
@@ -305,6 +304,7 @@ class TestMbFindSimilarRecordings:
                 "Tool", ["rock", "metal", "prog"],
             )
         call_args = mock_req.call_args
-        query = call_args[1]["query"] if "query" in (call_args[1] or {}) else call_args[0][1].get("query", "")
+        args = call_args[1] or {}
+        query = args.get("query") or call_args[0][1].get("query", "")
         # Verify the query contains both tag references
         assert "rock" in query or "metal" in query