diff --git a/.travis.yml b/.travis.yml
index b68e74c..fdd592e 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -19,8 +19,8 @@ install:
   - pip install poetry
   #- pip install tensorflow==$TF_VERSION
   # Work around https://github.com/python-poetry/poetry/issues/602
-  - poetry remove tensorflow
-  - poetry add tensorflow==$TF_VERSION
+  - poetry remove tensorflow tensorflow-estimator
+  - poetry add tensorflow==$TF_VERSION tensorflow-estimator==$TF_VERSION
   - poetry install -v
 
 script: pytest