diff --git a/src/f3dasm/_src/datageneration/datagenerator.py b/src/f3dasm/_src/datageneration/datagenerator.py
index 7c2d18be..b5cf2803 100644
--- a/src/f3dasm/_src/datageneration/datagenerator.py
+++ b/src/f3dasm/_src/datageneration/datagenerator.py
@@ -8,7 +8,7 @@
 # Standard
 import sys
 from functools import partial
-from typing import Callable
+from typing import Any, Callable
 
 if sys.version_info < (3, 8):  # NOQA
     from typing_extensions import Protocol  # NOQA
@@ -28,8 +28,15 @@
 
 
 class ExperimentSample(Protocol):
-    ...
+    def get(self, key: str) -> Any:
+        ...
+
+    def store(object: Any, name: str, to_disk: bool) -> None:
+        ...
 
+    @property
+    def job_number(self) -> int:
+        ...
 
 class DataGenerator:
     """Base class for a data generator"""