diff --git a/playground.py b/playground.py
index 779746e..5bfc764 100644
--- a/playground.py
+++ b/playground.py
@@ -762,7 +762,12 @@ for i, tracking in enumerate(trackings):
         tracking_2d_projection = camera.project(tracking.keypoints)
         for det in detections:
             delta_t = det.timestamp - tracking.last_active_timestamp
-            distance_2d = calculate_distance_2d(tracking_2d_projection, det.keypoints)
+            w, h = camera.params.image_size
+            distance_2d = calculate_distance_2d(
+                tracking_2d_projection,
+                det.keypoints,
+                image_size=(w, h),
+            )
             affinity_2d = calculate_affinity_2d(
                 distance_2d,
                 delta_t,