2 files changed, 2 insertions, 2 deletions

diff --git a/test/clitest.py b/test/clitest.py
index 453333a..781fb80 100644
--- a/test/clitest.py
+++ b/test/clitest.py
@@ -8,7 +8,7 @@ import subprocess
 import sys
 
 sys.path.append('..')
-from lib import mat
+from mat import mat
 import test
 
 class TestRemovecli(test.MATTest):
diff --git a/test/libtest.py b/test/libtest.py
index 2390841..d8aaac1 100644
--- a/test/libtest.py
+++ b/test/libtest.py
@@ -8,7 +8,7 @@ import unittest
 import test
 import sys
 sys.path.append('..')
-from lib import mat
+from mat import mat
 
 class TestRemovelib(test.MATTest):
     '''