diff --git a/tests/test_gis.py b/tests/test_gis.py
index f39554e..66d095c 100644
--- a/tests/test_gis.py
+++ b/tests/test_gis.py
@@ -108,7 +108,9 @@ def test_query_load_extension(use_spatialite_shortcut):
         ],
     )
     assert result.exit_code == 0, result.stdout
-    assert ["spatialite_version()"] == list(json.loads(result.output)[0].keys())
+    rows = json.loads(result.output)
+    assert len(rows) == 1
+    assert ["spatialite_version()"] == list(rows[0].keys())
 
 
 def test_cli_create_spatialite(tmpdir):