diff --git a/core/tests/test_type_util.py b/core/tests/test_type_util.py
index c8df58b..3d536cc 100644
--- a/core/tests/test_type_util.py
+++ b/core/tests/test_type_util.py
@@ -1,5 +1,5 @@
 from nose.tools import assert_equal
-from pydatasnippets.core.type_util import TypeUtil
+from ..type_util import TypeUtil
 
 
 class TestUtil():
diff --git a/core/unit_tests.ipynb b/core/unit_tests.ipynb
index b39e91e..a33b2f4 100644
--- a/core/unit_tests.ipynb
+++ b/core/unit_tests.ipynb
@@ -1,7 +1,7 @@
 {
  "metadata": {
   "name": "",
-  "signature": "sha256:bdca080b7a3c3e13406cff167e5e8a56e199b53e7c3b965c6583665afd737e4e"
+  "signature": "sha256:b98de97a0c9883157d25cd7ed2b6345daca13d7a67bed9ec4a72b8bec85d671b"
  },
  "nbformat": 3,
  "nbformat_minor": 0,
@@ -94,7 +94,7 @@
      "input": [
       "%%file tests/test_type_util.py\n",
       "from nose.tools import assert_equal\n",
-      "from pydatasnippets.core.type_util import TypeUtil\n",
+      "from ..type_util import TypeUtil\n",
       "\n",
       "\n",
       "class TestUtil():\n",