diff --git a/tests/workflow/test_optimized_workflow.py b/tests/workflow/test_optimized_workflow.py
index c2548d7..c331bf5 100644
--- a/tests/workflow/test_optimized_workflow.py
+++ b/tests/workflow/test_optimized_workflow.py
@@ -9,6 +9,7 @@
 from spatialyze.geospatial_video import GeospatialVideo
 from spatialyze.road_network import RoadNetwork
 from spatialyze.video_processor.camera_config import camera_config
+from spatialyze.video_processor.stages.tracking_3d.tracking_3d import Tracking3DResult
 from spatialyze.world import World, _execute
 from spatialyze.video_processor.cache import disable_cache
 from spatialyze.video_processor.metadata_json_encoder import MetadataJSONEncoder