diff --git a/src/pymovements/plotting/traceplot.py b/src/pymovements/plotting/traceplot.py
index e82e9fff1..9b6b62988 100644
--- a/src/pymovements/plotting/traceplot.py
+++ b/src/pymovements/plotting/traceplot.py
@@ -28,7 +28,7 @@
 from matplotlib import colors
 from matplotlib.collections import LineCollection
 
-from pymovements.gaze import GazeDataFrame
+from pymovements.gaze.gaze_dataframe import GazeDataFrame
 
 
 # This is really a dirty workaround to use the Agg backend if runnning pytest.