diff --git a/tests/integration/test_simple.py b/tests/integration/test_simple.py
index c4024914..007338c4 100644
--- a/tests/integration/test_simple.py
+++ b/tests/integration/test_simple.py
@@ -77,7 +77,7 @@ def test_predefined_models(
     }
     model = LuxonisModel(config_file, opts)
     model.train()
-    model.test()
+    model.test(view="train")
 
 
 def test_multi_input(opts: dict[str, Any], infer_path: Path):