diff --git a/heatmap_test.py b/heatmap_test.py
index 2c33894..83447ad 100644
--- a/heatmap_test.py
+++ b/heatmap_test.py
@@ -1,6 +1,6 @@
 import cv2
 from video_loader import VideoLoader
-from background_heatmap import BackgroundHeatmap, BackgroundHeatmapFromGroup
+from background_heatmap import BackgroundHeatmap
 import numpy as np