diff --git a/Makefile b/Makefile
index eec0ce82b..ed7ecfa56 100644
--- a/Makefile
+++ b/Makefile
@@ -72,5 +72,5 @@ repl:
 # Note: The pytest-randomly seed is fixed at 0 for now. Once the legacy np.random.seed(0)'s
 # are removed from tests, this can be removed, allowing all tests to use random seeds.
 test:
-	TF_USE_KERAS_LEGACY=1 pytest --randomly-seed=0 alibi_detect/saving/_tensorflow/tests/test_saving_tf.py
-	pytest --randomly-seed=0 alibi_detect --ignore alibi_detect/saving/_tensorflow/tests/test_saving_tf.py
\ No newline at end of file
+	TF_USE_KERAS_LEGACY=1 pytest --randomly-seed=0 alibi_detect/utils/tests/test_saving_legacy.py
+	pytest --randomly-seed=0 --ignore=alibi_detect/utils/tests/test_saving_legacy.py alibi_detect
\ No newline at end of file