diff --git a/generator/ctrl/install_ctrl.sh b/generator/ctrl/install_ctrl.sh
index 4a66e63..bc556ba 100755
--- a/generator/ctrl/install_ctrl.sh
+++ b/generator/ctrl/install_ctrl.sh
@@ -8,7 +8,7 @@ pip install tensorflow-gpu==1.14
 pip install keras
 
 # Patch the TensorFlow estimator package
-FILE="/usr/local/lib/python3.6/dist-packages/tensorflow_estimator/python/estimator/keras.py"
+FILE="/usr/local/lib/python2.7/dist-packages/tensorflow_estimator/python/estimator"
 patch -b "$FILE" estimator.patch
 
 # Install fastBPE