tests: Use wal functions.

1 files changed, 2 insertions, 2 deletions

diff --git a/tests/test_template.py b/tests/test_template.py
index 9040348..992c9c0 100755
--- a/tests/test_template.py
+++ b/tests/test_template.py
@@ -2,7 +2,7 @@
 import unittest
 import pathlib
 
-from pywal import template
+from pywal import wal
 from pywal import util
 
 
@@ -21,7 +21,7 @@ class TestExportColors(unittest.TestCase):
 
         output_dir = pathlib.Path("/tmp")
         template_dir = pathlib.Path("tests/test_files/templates")
-        template.export_all_templates(COLORS, output_dir, template_dir)
+        wal.export_all_templates(COLORS, output_dir, template_dir)
 
         result = pathlib.Path("/tmp/test_template").is_file()
         self.assertTrue(result)