diff --git a/src/visualise/__init__.py b/src/visualise/__init__.py
index 4ce0db5..84416b2 100644
--- a/src/visualise/__init__.py
+++ b/src/visualise/__init__.py
@@ -115,6 +115,7 @@ def gather_aoa(max_tof: float):
         while (datetime.now() - prev_frame).total_seconds() < 1 / config.HEATMAP_FPS:
             time.sleep(0.01)
             pass
+        prev_frame = datetime.now()
         logger.debug("Generating heatmap")
         buf = make_heatmap(max_tof)
         yield (b"--frame\r\nContent-Type: image/jpeg\r\n\r\n" + buf.read() + b"\r\n")