diff --git a/docker/github-worker-cuda-12-1/Dockerfile b/docker/github-worker-cuda-12-1/Dockerfile
index 93f5ecf3..2c623eb9 100644
--- a/docker/github-worker-cuda-12-1/Dockerfile
+++ b/docker/github-worker-cuda-12-1/Dockerfile
@@ -1,5 +1,5 @@
 # Temporary, until the rwkv package is public
-FROM ghcr.io/rwkv/rwkv-lm-lora:env-cuda-12-1
+FROM ghcr.io/RWKV/RWKV-infctx-trainer:env-cuda-12-1
 # FROM ghcr.io/rwkv/rwkv-infctx-trainer:env-cuda-12-1
 
 # Install the github runner