diff --git a/labbookdb/tests/test_query.py b/labbookdb/tests/test_query.py
index a9238fe..1d7911a 100644
--- a/labbookdb/tests/test_query.py
+++ b/labbookdb/tests/test_query.py
@@ -5,8 +5,8 @@
 def test_related_entry_separators():
 	session, engine = add.load_session("~/somepath.db")
 	with pytest.raises(Exception) as excinfo:
-		add.get_related_ids(s,e,"Animal:external_ids.AnimalExternalIdentifier:database.ETH/AIC/cdb&#&identifier.275511")
+		add.get_related_ids(session, engine, "Animal:external_ids.AnimalExternalIdentifier:database.ETH/AIC/cdb&#&identifier.275511")
 	assert excinfo.value.args[0] == 'No entry was found with a value of "275511" on the "identifier" column of the "AnimalExternalIdentifier" CATEGORY, in the database.'
 	with pytest.raises(Exception) as excinfo:
-		add.get_related_ids(s,e,"Animal:external_ids.AnimalExternalIdentifier:database.ETH/AIC/cdb&&identifier.275511")
+		add.get_related_ids(session, engine, "Animal:external_ids.AnimalExternalIdentifier:database.ETH/AIC/cdb&&identifier.275511")
 	assert excinfo.value.args[0] == 'No entry was found with a value of "ETH/AIC/cdb" on the "database" column of the "AnimalExternalIdentifier" CATEGORY, in the database.'