diff --git a/test/units/modules/network/check_point/test_cp_mgmt_address_range.py b/test/units/modules/network/check_point/test_cp_mgmt_address_range.py
index 72def6f0b9a..49d17d37498 100644
--- a/test/units/modules/network/check_point/test_cp_mgmt_address_range.py
+++ b/test/units/modules/network/check_point/test_cp_mgmt_address_range.py
@@ -20,10 +20,9 @@ from __future__ import absolute_import, division, print_function
 __metaclass__ = type
 
 import pytest
-from units.modules.utils import set_module_args, exit_json, fail_json, AnsibleFailJson, AnsibleExitJson
+from units.modules.utils import set_module_args, exit_json, fail_json, AnsibleExitJson
 
 from ansible.module_utils import basic
-from ansible.module_utils.network.checkpoint.checkpoint import api_call
 from ansible.modules.network.check_point import cp_mgmt_address_range
 
 function_path = 'ansible.modules.network.check_point.cp_mgmt_address_range.api_call'
diff --git a/test/units/modules/network/check_point/test_cp_mgmt_network.py b/test/units/modules/network/check_point/test_cp_mgmt_network.py
index 1f6eefdc823..d9fdd5be8f0 100644
--- a/test/units/modules/network/check_point/test_cp_mgmt_network.py
+++ b/test/units/modules/network/check_point/test_cp_mgmt_network.py
@@ -22,7 +22,6 @@ import pytest
 from units.modules.utils import set_module_args, exit_json, fail_json, AnsibleFailJson, AnsibleExitJson
 
 from ansible.module_utils import basic
-from ansible.module_utils.network.checkpoint.checkpoint import api_call
 from ansible.modules.network.check_point import cp_mgmt_network
 
 OBJECT = {'name': 'test_network', 'nat_settings': {'auto_rule': True,