diff --git a/.travis.yml b/.travis.yml
index 6df11b3..81faa10 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -9,11 +9,12 @@ before_install:
 
 install:
   # Get Migen / LiteX / Cores
+  - cd ~/
   - wget https://raw.githubusercontent.com/enjoy-digital/litex/master/litex_setup.py
   - python3 litex_setup.py init install
-  # litex_setup.py fails for litedram, which has already been cloned by Travis
-  # now install the local litedram so that it is available for benchmarks script
-  - python3 -m pip install .
+  # Install the version being tested
+  - cd $TRAVIS_BUILD_DIR
+  - python3 setup.py install
 
 before_script:
   # Get RISC-V toolchain
@@ -26,7 +27,9 @@ script: ./.sim-test.py --sdram-module="$SDRAM_MODULE"
 jobs:
   include:
     - stage: "Unit tests"
-      script: python setup.py test
+      script:
+        - cd $TRAVIS_BUILD_DIR
+        - python setup.py test
 
     - stage: "Simulations"
       env: SDRAM_MODULE=IS42S16160