diff --git a/tests/test_pkgset_source_koji.py b/tests/test_pkgset_source_koji.py
index 56c4ea71..61b45700 100644
--- a/tests/test_pkgset_source_koji.py
+++ b/tests/test_pkgset_source_koji.py
@@ -8,6 +8,8 @@ import six
 from ddt import ddt, data, unpack
 from typing import AnyStr, List, Set, Dict, Tuple
 
+from tests.test_gather_method_hybrid import MockModule
+
 try:
     import unittest2 as unittest
     from unittest2 import mock