diff --git a/ml_experiment/ExperimentDefinition.py b/ml_experiment/ExperimentDefinition.py
index bfc1cc7..6f7dad3 100644
--- a/ml_experiment/ExperimentDefinition.py
+++ b/ml_experiment/ExperimentDefinition.py
@@ -5,7 +5,7 @@
 from ml_experiment.metadata.MetadataTable import MetadataTable
 
 class ExperimentDefinition:
-    def __init__(self, part_name: str, version: int | None, base: str | None = None):
+    def __init__(self, part_name: str, version: int, base: str | None = None):
         self.part_name = part_name
         self.version = version
         self.base_path = base or os.getcwd()