diff --git a/requirements.txt b/requirements.txt
index 21ad8f7..2a605ff 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,11 +1,11 @@
 torch
-gymnasium
+gymnasium==1.2.3
 hydra-core
 omegaconf
-mujoco
-mujoco-mjx
-jax
-skrl[torch]
+mujoco==3.5.0
+mujoco-mjx==3.5.0
+jax==0.9.1
+skrl[torch]==1.4.3
 clearml
 imageio
 imageio-ffmpeg
diff --git a/scripts/train.py b/scripts/train.py
index 080db3f..2ca3867 100644
--- a/scripts/train.py
+++ b/scripts/train.py
@@ -71,11 +71,11 @@ def _init_clearml(choices: dict[str, str], remote: bool = False) -> Task:
 
     task = Task.init(project_name=project, task_name=task_name, tags=tags)
     task.set_base_docker(
-        "registry.kube.optimize/worker-image:latest",
+        "git.victormylle.be/victormylle/simple-rl-framework:latest",
         docker_setup_bash_script=(
             "apt-get update && apt-get install -y --no-install-recommends "
             "libosmesa6-dev libgl1-mesa-glx libglfw3 && rm -rf /var/lib/apt/lists/* "
-            "&& pip install 'jax[cuda12]' mujoco-mjx PyOpenGL PyOpenGL-accelerate"
+            "&& pip install 'jax[cuda12]==0.9.1' mujoco-mjx==3.5.0 PyOpenGL PyOpenGL-accelerate"
         ),
         docker_arguments=[
             "-e", "MUJOCO_GL=osmesa",