diff --git a/.travis.yml b/.travis.yml
index 3ca3044..850b4e8 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -9,4 +9,4 @@ install:
     - "pip install pyfakefs"
     - "python setup.py install"
 # command to run tests
-script: nosetests
+script: python -m unittest discover
diff --git a/tests/__init__.py b/tests/__init__.py
index e69de29..6acb07d 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -0,0 +1,4 @@
+import sys, os
+
+this_dir = os.path.dirname(os.path.abspath(__file__))
+sys.path.insert(0, this_dir)