diff --git a/Dockerfile b/Dockerfile
index 71a833c..cd5bc39 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM gcr.io/tensorflow/tensorflow:1.3.0
+FROM tensorflow/tensorflow:1.3.0
 
 RUN pip install networkx==1.11
 RUN rm /notebooks/*
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 681f22c..cf1eb38 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM gcr.io/tensorflow/tensorflow:1.3.0-gpu
+FROM tensorflow/tensorflow:1.3.0-gpu
 
 RUN pip install networkx==1.11
 RUN rm /notebooks/*