diff --git a/test/units/modules/core/test_apt.py b/test/units/modules/core/test_apt.py
index bd43a4aeae..47e4132797 100644
--- a/test/units/modules/core/test_apt.py
+++ b/test/units/modules/core/test_apt.py
@@ -2,7 +2,8 @@ import collections
 import mock
 import os
 import sys
-import unittest
+from ansible.compat.tests import unittest
+
 
 try:
     from ansible.modules.core.packaging.os.apt import (