diff --git a/setup.py b/setup.py
index <HASH>..<HASH> 100755
--- a/setup.py
+++ b/setup.py
@@ -40,7 +40,9 @@ class PyTest(TestCommand):
         # import here, cause outside the eggs aren't loaded
         import pytest
         test_path = path.join(path.dirname(path.abspath(__file__)), 'source', 'test')
+        rafcon_path = path.join(path.dirname(path.abspath(__file__)), 'source')
         sys.path.insert(0, test_path)
+        sys.path.insert(0, rafcon_path)
         error_number = pytest.main(self.pytest_args + ['source/test/all'])
         sys.exit(error_number)
Tests: Add RAFCON path to PYTHONPATH