diff --git a/.travis.yml b/.travis.yml
index d47f4c5..c4e1742 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -3,6 +3,6 @@ python:
   - "2.7"
 install:
 - pip install coveralls
-script: coverage run --source wrc -m py.test tests.py -v
+script: coverage run --source WazeRouteCalculator -m py.test tests.py -v
 after_success:
   coveralls
diff --git a/WazeRouteCalculator/__init__.py b/WazeRouteCalculator/__init__.py
index c281e62..75bae6e 100644
--- a/WazeRouteCalculator/__init__.py
+++ b/WazeRouteCalculator/__init__.py
@@ -1,2 +1,2 @@
 from .__version__ import __version__
-from .WazeRouteCalculator import WazeRouteCalculator, WRCError
+from .WazeRouteCalculator import *
diff --git a/tests.py b/tests.py
index 0cc0356..580c3cf 100644
--- a/tests.py
+++ b/tests.py
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-import wrc.wrc as wrc
+import WazeRouteCalculator as wrc
 import mock
 import StringIO