diff --git a/tests/test_cli.py b/tests/test_cli.py
index 335084f..abba431 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -4,8 +4,7 @@ import unittest
 import git
 from click.testing import CliRunner
 
-import changelog
-import yaclog
+import yaclog.changelog
 from yaclog.cli.__main__ import cli
 
 
@@ -66,7 +65,7 @@ class TestTagging(unittest.TestCase):
 
         with runner.isolated_filesystem():
             in_log = yaclog.Changelog(location)
-            in_log.versions = [changelog.VersionEntry(), changelog.VersionEntry()]
+            in_log.versions = [yaclog.changelog.VersionEntry(), yaclog.changelog.VersionEntry()]
 
             in_log.versions[0].name = '1.0.0'
             in_log.versions[1].name = '0.9.0'
@@ -94,7 +93,7 @@ class TestTagging(unittest.TestCase):
         with runner.isolated_filesystem():
             in_log = yaclog.Changelog(location)
             in_log.versions = [None, None]
-            in_log.versions = [changelog.VersionEntry(), changelog.VersionEntry()]
+            in_log.versions = [yaclog.changelog.VersionEntry(), yaclog.changelog.VersionEntry()]
 
             in_log.versions[0].name = '1.0.0'
             in_log.versions[0].tags = ['TAG1']