diff --git a/opensfm/test/test_commands.py b/opensfm/test/test_commands.py
index a5890bcd9..ac04e8602 100644
--- a/opensfm/test/test_commands.py
+++ b/opensfm/test/test_commands.py
@@ -3,7 +3,7 @@
 from os.path import join
 
 from opensfm import commands, dataset
-from opensfm.test import data_generation, utils
+from mapillary.opensfm.opensfm.test import data_generation, utils
 
 
 def run_command(command, args) -> None:
diff --git a/opensfm/test/test_dataset.py b/opensfm/test/test_dataset.py
index 60022ae0a..142887c82 100644
--- a/opensfm/test/test_dataset.py
+++ b/opensfm/test/test_dataset.py
@@ -1,7 +1,7 @@
 # pyre-unsafe
 import numpy as np
 from opensfm import features
-from opensfm.test import data_generation
+from mapillary.opensfm.opensfm.test import data_generation
 
 
 def test_dataset_load_features_sift(tmpdir) -> None:
diff --git a/opensfm/test/test_datastructures.py b/opensfm/test/test_datastructures.py
index 5e9c5cd9a..cbf5116be 100644
--- a/opensfm/test/test_datastructures.py
+++ b/opensfm/test/test_datastructures.py
@@ -11,11 +11,7 @@
 from opensfm import pygeometry
 from opensfm import pymap
 from opensfm import types
-from opensfm.test.utils import (
-    assert_maps_equal, assert_metadata_equal,
-    assert_cameras_equal,
-    assert_shots_equal,
-)
+from mapillary.opensfm.opensfm.test.utils import assert_maps_equal, assert_metadata_equal, assert_cameras_equal, assert_shots_equal
 
 
 def _create_reconstruction(
diff --git a/opensfm/test/test_io.py b/opensfm/test/test_io.py
index 0b43af2c5..39e46890d 100644
--- a/opensfm/test/test_io.py
+++ b/opensfm/test/test_io.py
@@ -5,7 +5,7 @@
 
 import numpy as np
 from opensfm import io, pygeometry, types
-from opensfm.test import data_generation, utils
+from mapillary.opensfm.opensfm.test import data_generation, utils
 
 
 filename = os.path.join(
diff --git a/opensfm/test/test_pairs_selection.py b/opensfm/test/test_pairs_selection.py
index b16a2ce69..303f334ff 100644
--- a/opensfm/test/test_pairs_selection.py
+++ b/opensfm/test/test_pairs_selection.py
@@ -6,7 +6,7 @@
 import numpy as np
 import pytest
 from opensfm import commands, dataset, feature_loader, pairs_selection, geo
-from opensfm.test import data_generation
+from mapillary.opensfm.opensfm.test import data_generation
 from opensfm.dataset_base import DataSetBase