diff --git a/dkg/Dockerfile b/dkg/Dockerfile
index 4ee42c0..4728f72 100644
--- a/dkg/Dockerfile
+++ b/dkg/Dockerfile
@@ -1,5 +1,5 @@
-ARG DKG_UPSTREAM_VERSION
-FROM ssvlabs/ssv-dkg:${DKG_UPSTREAM_VERSION}
+ARG DKG_IMAGE
+FROM ${DKG_IMAGE}
 
 WORKDIR /
 
diff --git a/docker-compose.yml b/docker-compose.yml
index f2ae940..1630134 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -20,7 +20,7 @@ services:
     build:
       context: dkg
       args:
-        DKG_UPSTREAM_VERSION: v3.1.0
+        DKG_IMAGE: public.ecr.aws/h4h2r1n7/dkg:v3.1.1
         STAKER_SCRIPTS_VERSION: v0.1.2
     restart: on-failure
     volumes: