diff --git a/.travis.yml b/.travis.yml
index d24c3ad2d8..3e4971d157 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -49,7 +49,6 @@ before_install:
 install:
   - conda install --quiet --yes pip python=$TRAVIS_PYTHON_VERSION numpy scipy matplotlib cython ipython h5py;
   - pip install -r requirements_dev.txt
-  - pip install git+https://github.com/simpeg/discretize.git@repo-cleanup
   - python setup.py install
   - if [ "$TEST_DIR" = "$DEPLOY_DIR" ]; then
       if [ ! -d ${HOME}/google-cloud-sdk ]; then curl https://sdk.cloud.google.com | bash; fi ;
diff --git a/setup.py b/setup.py
index 158e1bcaff..f55d1f2109 100644
--- a/setup.py
+++ b/setup.py
@@ -41,7 +41,7 @@
         'matplotlib',
         'properties>=0.5.2',
         'vectormath>=0.2.0',
-        'discretize>=0.2.0',
+        'discretize>=0.4.0',
         'geoana>=0.0.4'
     ],
     author="Rowan Cockett",