diff --git a/src/bonsai/test/core/test_georeference.py b/src/bonsai/test/core/test_georeference.py
index b9b330d816..e95236ec3f 100644
--- a/src/bonsai/test/core/test_georeference.py
+++ b/src/bonsai/test/core/test_georeference.py
@@ -23,6 +23,7 @@ from test.core.bootstrap import georeference, ifc
 class TestAddGeoreferencing:
     def test_run(self, georeference):
         georeference.add_georeferencing().should_be_called()
+        georeference.set_model_origin().should_be_called()
         subject.add_georeferencing(georeference)
 
 
@@ -35,9 +36,10 @@ class TestEnableEditingGeoreferencing:
 
 
 class TestRemoveGeoreferencing:
-    def test_run(self, ifc):
+    def test_run(self, ifc, georeference):
         ifc.run("georeference.remove_georeferencing").should_be_called()
-        subject.remove_georeferencing(ifc)
+        georeference.set_model_origin().should_be_called()
+        subject.remove_georeferencing(ifc, georeference)
 
 
 class TestDisableEditingGeoreferencing: