diff --git a/.github/workflows/docker-build-action.yaml b/.github/workflows/docker-build-action.yaml
index 41d37d47..3bd48a88 100644
--- a/.github/workflows/docker-build-action.yaml
+++ b/.github/workflows/docker-build-action.yaml
@@ -36,11 +36,9 @@ jobs:
         uses: actions/checkout@v3
 
       - name: Tailscale
-        uses: tailscale/github-action@v1
+        uses: huggingface/tailscale-action@main
         with:
           authkey: ${{ secrets.TAILSCALE_AUTHKEY }}
-          version: 1.38.2
-
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v2.0.0
         with:
@@ -63,4 +61,4 @@ jobs:
           context: ${{ inputs.context }}
           build-args: ${{ inputs.build_args }}
           file:  ${{ inputs.context }}/${{ inputs.dockerfile }}
-          tags: ${{ inputs.repository }}/${{ inputs.image }}:${{ env.GITHUB_SHA_SHORT }},${{ inputs.repository }}/${{ inputs.image }}:latest
\ No newline at end of file
+          tags: ${{ inputs.repository }}/${{ inputs.image }}:${{ env.GITHUB_SHA_SHORT }},${{ inputs.repository }}/${{ inputs.image }}:latest
diff --git a/dockerfiles/tensorflow/gpu/Dockerfile b/dockerfiles/tensorflow/gpu/Dockerfile
index 02018371..c42a33c0 100644
--- a/dockerfiles/tensorflow/gpu/Dockerfile
+++ b/dockerfiles/tensorflow/gpu/Dockerfile
@@ -34,11 +34,6 @@ RUN curl -L https://micro.mamba.pm/api/micromamba/linux-64/latest | tar -xj "bin
 
 WORKDIR /app
 
-RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
-    source $HOME/.cargo/env && \
-    source .venv/bin/activate && \
-    ls -all
-
 # install base python dependencies
 COPY dockerfiles/tensorflow/gpu/environment.yaml /app/environment.yaml
 RUN micromamba install -y -n base -f environment.yaml \