diff --git a/src/pymovements/__init__.py b/src/pymovements/__init__.py
index e3fff1d48..d7591775a 100644
--- a/src/pymovements/__init__.py
+++ b/src/pymovements/__init__.py
@@ -34,9 +34,9 @@
 from pymovements.events import EventDataFrame
 from pymovements.events import EventGazeProcessor
 from pymovements.events import EventProcessor
-from pymovements.gaze.experiment import Experiment
-from pymovements.gaze.gaze_dataframe import GazeDataFrame
-from pymovements.gaze.screen import Screen
+from pymovements.gaze import Experiment
+from pymovements.gaze import GazeDataFrame
+from pymovements.gaze import Screen
 
 
 __all__ = [
diff --git a/src/pymovements/events/processing.py b/src/pymovements/events/processing.py
index 15a51bf86..3abf2c9f8 100644
--- a/src/pymovements/events/processing.py
+++ b/src/pymovements/events/processing.py
@@ -26,10 +26,10 @@
 
 import polars as pl
 
+import pymovements as pm
 from pymovements.events.frame import EventDataFrame
 from pymovements.events.properties import EVENT_PROPERTIES
 from pymovements.exceptions import InvalidProperty
-from pymovements.gaze.gaze_dataframe import GazeDataFrame
 
 
 class EventProcessor:
@@ -140,7 +140,7 @@ def __init__(
     def process(
             self,
             events: EventDataFrame,
-            gaze: GazeDataFrame,
+            gaze: pm.GazeDataFrame,
             identifiers: str | list[str],
             name: str | None = None,
     ) -> pl.DataFrame: