diff --git a/src/pymovements/dataset/dataset.py b/src/pymovements/dataset/dataset.py
index 159f7f2f9..987d39a2f 100644
--- a/src/pymovements/dataset/dataset.py
+++ b/src/pymovements/dataset/dataset.py
@@ -35,7 +35,7 @@
 from pymovements.dataset.dataset_paths import DatasetPaths
 from pymovements.events.frame import EventDataFrame
 from pymovements.events.processing import EventGazeProcessor
-from pymovements.gaze.gaze_dataframe import GazeDataFrame
+from pymovements.gaze import GazeDataFrame
 
 
 class Dataset:
diff --git a/src/pymovements/plotting/heatmap.py b/src/pymovements/plotting/heatmap.py
index 30574efe7..44da8953a 100644
--- a/src/pymovements/plotting/heatmap.py
+++ b/src/pymovements/plotting/heatmap.py
@@ -24,7 +24,7 @@
 import numpy as np
 from matplotlib import colors
 
-from pymovements.gaze.gaze_dataframe import GazeDataFrame
+from pymovements.gaze import GazeDataFrame
 
 
 def heatmap(
diff --git a/src/pymovements/plotting/main_sequence_plot.py b/src/pymovements/plotting/main_sequence_plot.py
index f19aae71e..551e73152 100644
--- a/src/pymovements/plotting/main_sequence_plot.py
+++ b/src/pymovements/plotting/main_sequence_plot.py
@@ -25,7 +25,7 @@
 from matplotlib.collections import Collection
 from polars import ColumnNotFoundError
 
-from pymovements.events.frame import EventDataFrame
+from pymovements.events import EventDataFrame
 
 
 def main_sequence_plot(
diff --git a/src/pymovements/plotting/traceplot.py b/src/pymovements/plotting/traceplot.py
index 9b6b62988..e82e9fff1 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.gaze_dataframe import GazeDataFrame
+from pymovements.gaze import GazeDataFrame
 
 
 # This is really a dirty workaround to use the Agg backend if runnning pytest.
diff --git a/src/pymovements/plotting/tsplot.py b/src/pymovements/plotting/tsplot.py
index 767545bc0..8207ea715 100644
--- a/src/pymovements/plotting/tsplot.py
+++ b/src/pymovements/plotting/tsplot.py
@@ -26,7 +26,7 @@
 import numpy as np
 import polars as pl
 
-from pymovements.gaze.gaze_dataframe import GazeDataFrame
+from pymovements.gaze import GazeDataFrame
 
 
 def tsplot(