diff --git a/neural-networks/object-tracking/kalman/utils/kalman_filter_node.py b/neural-networks/object-tracking/kalman/utils/kalman_filter_node.py
index ecd107382..a224e281b 100644
--- a/neural-networks/object-tracking/kalman/utils/kalman_filter_node.py
+++ b/neural-networks/object-tracking/kalman/utils/kalman_filter_node.py
@@ -90,6 +90,8 @@ def process(self, img_frame: dai.Buffer, tracklets: dai.Buffer) -> None:
 
                 vec_bbox = self._kalman_filters[t.id]["bbox"].x
                 vec_space = self._kalman_filters[t.id]["space"].x
+                vec_bbox = vec_bbox.flatten()
+                vec_space = vec_space.flatten()
 
                 x1_filter = (vec_bbox[0] - vec_bbox[2] / 2) / img_frame.getWidth()
                 x2_filter = (vec_bbox[0] + vec_bbox[2] / 2) / img_frame.getWidth()