diff --git a/pyproject.toml b/pyproject.toml
index 7ad038c006e314c3385c575bf152be9415194bc0..745d0473337e454a3148a397a939af94af7d0949 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,10 +8,10 @@ authors = [
 ]
 requires-python = ">=3.10"
 dependencies = [
-    "artistlib ; sys_platform == 'windows'",
+    "artistlib ; sys_platform == 'win32'",
     "dm-pix>=0.4.3",
     "h5py>=3.12.1",
-    "jax>=0.4.35 ; sys_platform == 'windows'",
+    "jax>=0.4.35 ; sys_platform == 'win32'",
     "jax[cuda12]>=0.4.35 ; sys_platform == 'linux'",
     "matplotlib>=3.9.2",
     "numpy>=2.1.3",
diff --git a/src/cyxtrax/simulation/model/augemtation_parameters.py b/src/cyxtrax/simulation/model/augemtation_parameters.py
index e478c5b23df199a9ed491e4adaec7cd11410a8ba..d3514b192848c7668ba8401a7f2215c9e39444a5 100644
--- a/src/cyxtrax/simulation/model/augemtation_parameters.py
+++ b/src/cyxtrax/simulation/model/augemtation_parameters.py
@@ -7,3 +7,11 @@ class AugemntationParameter:
     augmentation_function: str
     function_parameter: dict
     input_parameter: dict
+
+    def as_dict(self):
+        return dict(
+            augmentation_module=self.augmentation_module,
+            augmentation_function=self.augmentation_function,
+            function_parameter=self.function_parameter,
+            input_parameter=self.input_parameter,
+        )
diff --git a/uv.lock b/uv.lock
index 05ae03d7f242ccd26ad8d32bebea9384e277b46c..2b921bf9b5ceeaf450fffc5aa73b9fd2e46a314f 100644
--- a/uv.lock
+++ b/uv.lock
@@ -123,7 +123,7 @@ name = "cyxtrax"
 version = "0.1.0"
 source = { editable = "." }
 dependencies = [
-    { name = "artistlib", marker = "sys_platform == 'windows'" },
+    { name = "artistlib", marker = "sys_platform == 'win32'" },
     { name = "dm-pix" },
     { name = "h5py" },
     { name = "jax", marker = "sys_platform == 'linux' or sys_platform == 'windows'" },
@@ -137,7 +137,7 @@ dependencies = [
 
 [package.metadata]
 requires-dist = [
-    { name = "artistlib", marker = "sys_platform == 'windows'", git = "https://github.com/wittlsn/aRTist-PythonLib" },
+    { name = "artistlib", marker = "sys_platform == 'win32'", git = "https://github.com/wittlsn/aRTist-PythonLib" },
     { name = "dm-pix", specifier = ">=0.4.3" },
     { name = "h5py", specifier = ">=3.12.1" },
     { name = "jax", marker = "sys_platform == 'windows'", specifier = ">=0.4.35" },