tests: Fix tests.

2 files changed, 5 insertions, 5 deletions

diff --git a/tests/test_colors.py b/tests/test_image.py
index 077600e..7dacecb 100644
--- a/tests/test_colors.py
+++ b/tests/test_image.py
@@ -1,19 +1,19 @@
 """Test image functions."""
 import unittest
 
-from pywal import image
+from pywal import wal
 
 
 class TestImage(unittest.TestCase):
     """Test image functions."""
     def test_get_img(self):
         """> Validate image file."""
-        result = image.get_image("tests/test_files/test.jpg")
+        result = wal.get_image("tests/test_files/test.jpg")
         self.assertEqual(result, "tests/test_files/test.jpg")
 
     def test_get_img_dir(self):
         """> Validate image directory."""
-        result = image.get_image("tests/test_files")
+        result = wal.get_image("tests/test_files")
         self.assertEqual(result, "tests/test_files/test.jpg")
 
 
diff --git a/tests/test_magic.py b/tests/test_magic.py
index 673420c..852f8b3 100755
--- a/tests/test_magic.py
+++ b/tests/test_magic.py
@@ -1,7 +1,7 @@
 """Test imagemagick functions."""
 import unittest
 
-from pywal import magic
+from pywal import wal
 
 
 class TestGenColors(unittest.TestCase):
@@ -9,7 +9,7 @@ class TestGenColors(unittest.TestCase):
 
     def test_gen_colors(self):
         """> Generate a colorscheme."""
-        result = magic.gen_colors("tests/test_files/test.jpg")
+        result = wal.create_palette("tests/test_files/test.jpg")
         self.assertEqual(result[0], "#0F191A")