diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index 3a19563..b91d81d 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -35,9 +35,9 @@ jobs:
           username: konstantinjdobler # change this
           password: ${{secrets.DOCKER_REGISTRY_TOKEN}}
       - name: Get torch version for amd
-        run: "echo TORCH_VERSION_AMD=$(.github/scripts/release.py pytorch) >> $GITHUB_ENV"
+        run: "echo TORCH_VERSION_AMD=$(.github/scripts/extract_version_info.py pytorch) >> $GITHUB_ENV"
       - name: Get cuda version for amd
-        run: "echo CUDA_VERSION_AMD=$(.github/scripts/release.py pytorch-cuda) >> $GITHUB_ENV"
+        run: "echo CUDA_VERSION_AMD=$(.github/scripts/extract_version_info.py pytorch-cuda) >> $GITHUB_ENV"
       - name: Print version info
         run: |
           echo "TORCH_VERSION_AMD=$TORCH_VERSION_AMD"
diff --git a/Dockerfile b/Dockerfile
index f414e78..e443a66 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -145,4 +145,4 @@ USER $MAMBA_USER
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN micromamba config prepend channels conda-forge --env
 # Disable micromamba banner at every command
-RUN micromamba config set show_banner false --env
\ No newline at end of file
+RUN micromamba config set show_banner false --env