diff --git a/spatialyze/video_processor/stages/segment_trajectory/__init__.py b/spatialyze/video_processor/stages/segment_trajectory/__init__.py
index a0d0f23..885b39f 100644
--- a/spatialyze/video_processor/stages/segment_trajectory/__init__.py
+++ b/spatialyze/video_processor/stages/segment_trajectory/__init__.py
@@ -1,4 +1,4 @@
-from typing import Any, Dict, NamedTuple, Tuple
+from typing import Any, Dict, Tuple
 
 from ...payload import Payload
 from ...types import DetectionId
diff --git a/spatialyze/video_processor/stages/segment_trajectory/construct_segment_trajectory.py b/spatialyze/video_processor/stages/segment_trajectory/construct_segment_trajectory.py
index a68bfae..0094db3 100644
--- a/spatialyze/video_processor/stages/segment_trajectory/construct_segment_trajectory.py
+++ b/spatialyze/video_processor/stages/segment_trajectory/construct_segment_trajectory.py
@@ -1,13 +1,14 @@
 import datetime
 import math
 from dataclasses import dataclass
-from typing import List, Tuple, Union
+from typing import List, NamedTuple, Tuple, Union
 
 import numpy as np
 import numpy.typing as npt
 import postgis
 import psycopg2
 import psycopg2.sql
+import shapely.geometry
 from plpygis import Geometry
 from shapely.geometry import Point
 from shapely.ops import nearest_points