diff --git a/install_gce.sh b/install_gce.sh
index 674888d..9e19148 100644
--- a/install_gce.sh
+++ b/install_gce.sh
@@ -10,7 +10,6 @@ sudo pip install Cython
 # Patch the TensorFlow estimator package
 FILE="/usr/local/lib/python2.7/dist-packages/tensorflow_estimator/python/estimator/keras.py"
 sudo patch -b "$FILE" estimator.patch
-sudo awk 'NR==24{print "import tensorflow as tf"}7' "$FILE" > /tmp/out && sudo mv /tmp/out "$FILE"
 
 # Install fastBPE
 git clone https://github.com/glample/fastBPE.git
@@ -18,6 +17,7 @@ cd fastBPE
 sudo python setup.py install
 cd ..
 
+# Download the 512-length model if specified, 256-length otherwise
 if [ "$1" = "512" ]
 then
     URL="gs://sf-ctrl/seqlen512_v1.ckpt/"
@@ -26,4 +26,4 @@ else
 fi
 
 # Copy model
-gsutil -m cp -r $URL .
+gsutil -m cp -r "$URL" .