diff --git a/pytorchvideo_trainer/tests/test_task_byol.py b/pytorchvideo_trainer/tests/test_task_byol.py
index b879ef1d..ecfa1768 100644
--- a/pytorchvideo_trainer/tests/test_task_byol.py
+++ b/pytorchvideo_trainer/tests/test_task_byol.py
@@ -2,7 +2,7 @@
 
 # pyre-strict
 from torchrecipes.core.base_train_app import BaseTrainApp
-from util import (
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
     BaseTrainAppTestCase,
     create_small_kinetics_dataset,
     run_locally,
diff --git a/pytorchvideo_trainer/tests/test_task_moco_v2.py b/pytorchvideo_trainer/tests/test_task_moco_v2.py
index adfbfed5..3eaeb9d6 100644
--- a/pytorchvideo_trainer/tests/test_task_moco_v2.py
+++ b/pytorchvideo_trainer/tests/test_task_moco_v2.py
@@ -2,7 +2,7 @@
 
 # pyre-strict
 from torchrecipes.core.base_train_app import BaseTrainApp
-from util import (
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
     BaseTrainAppTestCase,
     create_small_kinetics_dataset,
     run_locally,
diff --git a/pytorchvideo_trainer/tests/test_task_module_all.py b/pytorchvideo_trainer/tests/test_task_module_all.py
index 95f8c1a5..9542b05e 100644
--- a/pytorchvideo_trainer/tests/test_task_module_all.py
+++ b/pytorchvideo_trainer/tests/test_task_module_all.py
@@ -10,7 +10,11 @@
 from pytorch_lightning import Trainer
 from pytorchvideo_trainer.datamodule.datamodule import VideoClassificationDataModuleConf
 from pytorchvideo_trainer.train_app import VideoClassificationTrainAppConf
-from util import create_small_kinetics_dataset, run_locally, tempdir
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
+    create_small_kinetics_dataset,
+    run_locally,
+    tempdir,
+)
 
 
 class TestMain(unittest.TestCase):
diff --git a/pytorchvideo_trainer/tests/test_task_simclr.py b/pytorchvideo_trainer/tests/test_task_simclr.py
index 4ac47880..bf8c6a62 100644
--- a/pytorchvideo_trainer/tests/test_task_simclr.py
+++ b/pytorchvideo_trainer/tests/test_task_simclr.py
@@ -2,7 +2,7 @@
 
 # pyre-strict
 from torchrecipes.core.base_train_app import BaseTrainApp
-from util import (
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
     BaseTrainAppTestCase,
     create_small_kinetics_dataset,
     run_locally,
diff --git a/pytorchvideo_trainer/tests/test_task_video_classification.py b/pytorchvideo_trainer/tests/test_task_video_classification.py
index 685c6e1b..c45ef8d6 100644
--- a/pytorchvideo_trainer/tests/test_task_video_classification.py
+++ b/pytorchvideo_trainer/tests/test_task_video_classification.py
@@ -2,7 +2,7 @@
 
 # pyre-strict
 from torchrecipes.core.base_train_app import BaseTrainApp
-from util import (
+from vision.fair.pytorchvideo.pytorchvideo_trainer.tests.util import (
     BaseTrainAppTestCase,
     create_small_kinetics_dataset,
     run_locally,