diff --git a/.flexci/build_and_push.sh b/.flexci/build_and_push.sh
index 3e73b99..2e8a018 100644
--- a/.flexci/build_and_push.sh
+++ b/.flexci/build_and_push.sh
@@ -3,7 +3,7 @@
 IMAGE_BASE="${1:-}"
 IMAGE_PUSH=1
 if [ "${IMAGE_BASE}" = "" ]; then
-  IMAGE_BASE="torch-dftd3"
+  IMAGE_BASE="torch-dftd"
   IMAGE_PUSH=0
 fi
 
diff --git a/.flexci/pytest_script.sh b/.flexci/pytest_script.sh
index bff615b..f744410 100644
--- a/.flexci/pytest_script.sh
+++ b/.flexci/pytest_script.sh
@@ -2,7 +2,7 @@
 set -eu
 
 #IMAGE=pytorch/pytorch:1.5.1-cuda10.1-cudnn7-devel
-IMAGE=asia.gcr.io/pfn-public-ci/torch-dftd3-ci:torch15
+IMAGE=asia.gcr.io/pfn-public-ci/torch-dftd-ci:torch15
 
 
 main() {