diff --git a/test/test_automaton.py b/test/test_automaton.py
index 6ae308d..f917b9e 100644
--- a/test/test_automaton.py
+++ b/test/test_automaton.py
@@ -15,7 +15,7 @@ limitations under the License.
 """
 
 import sys
-sys.path.append('../..')
+sys.path.append('../cellular_automaton')
 
 from cellular_automaton import *
 import unittest
diff --git a/test/test_cell.py b/test/test_cell.py
index 90af91c..9f7dbcc 100644
--- a/test/test_cell.py
+++ b/test/test_cell.py
@@ -15,10 +15,10 @@ limitations under the License.
 """
 
 import sys
-sys.path.append('../..')
+sys.path.append('../cellular_automaton')
 
-from cellular_automaton.cellular_automaton.cell import Cell
-from cellular_automaton.cellular_automaton.cell_state import CellState
+from cellular_automaton.cell import Cell
+from cellular_automaton.cell_state import CellState
 import unittest
 
 
diff --git a/test/test_cell_state.py b/test/test_cell_state.py
index 03137d7..54d8415 100644
--- a/test/test_cell_state.py
+++ b/test/test_cell_state.py
@@ -15,9 +15,9 @@ limitations under the License.
 """
 
 import sys
-sys.path.append('../..')
+sys.path.append('../cellular_automaton')
 
-from cellular_automaton.cellular_automaton import cell_state as cell_state
+from cellular_automaton import cell_state
 import unittest
 
 
diff --git a/test/test_factory.py b/test/test_factory.py
index 3ca0479..5b9614a 100644
--- a/test/test_factory.py
+++ b/test/test_factory.py
@@ -15,11 +15,11 @@ limitations under the License.
 """
 
 import sys
-sys.path.append('../..')
+sys.path.append('../cellular_automaton')
 
-from cellular_automaton.cellular_automaton import *
-from cellular_automaton.cellular_automaton.cell_state import CellState
-from cellular_automaton.cellular_automaton.state import CellularAutomatonState
+from cellular_automaton import *
+from cellular_automaton.cell_state import CellState
+from cellular_automaton.state import CellularAutomatonState
 import unittest
 import mock
 
diff --git a/test/test_neighborhood.py b/test/test_neighborhood.py
index 3135913..2d7488c 100644
--- a/test/test_neighborhood.py
+++ b/test/test_neighborhood.py
@@ -15,9 +15,9 @@ limitations under the License.
 """
 
 import sys
-sys.path.append('../..')
+sys.path.append('../cellular_automaton')
 
-from cellular_automaton import cellular_automaton as csn
+import cellular_automaton as csn
 import unittest