1 files changed, 3 insertions, 7 deletions
diff --git a/tests/test_image.py b/tests/test_image.py
index 95a357e..da557a7 100644
--- a/tests/test_image.py
+++ b/tests/test_image.py
@@ -1,23 +1,19 @@
 """Test image functions."""
-import pathlib
 import unittest
 
-from pywal import wal
-
-
-CACHE_DIR = pathlib.Path("/tmp/wal")
+from pywal import image
 
 
 class TestImage(unittest.TestCase):
     """Test image functions."""
     def test_get_img(self):
         """> Validate image file."""
-        result = wal.get_image("tests/test_files/test.jpg", CACHE_DIR)
+        result = image.get("tests/test_files/test.jpg")
         self.assertEqual(result, "tests/test_files/test.jpg")
 
     def test_get_img_dir(self):
         """> Validate image directory."""
-        result = wal.get_image("tests/test_files", CACHE_DIR)
+        result = image.get("tests/test_files")
         self.assertEqual(result, "tests/test_files/test2.jpg")