diff --git a/tests/console/Policies.py b/tests/console/Policies.py
index ce7b769e94..1cf76c93e8 100644
--- a/tests/console/Policies.py
+++ b/tests/console/Policies.py
@@ -3,6 +3,7 @@ from include.common_classes_60 import PandoraWebDriverTestCase
 from include.common_functions_60 import login, detect_and_pass_all_wizards, gen_random_string, is_enterprise
 from include.policy_functions import *
 from include.agent_functions import *
+from include.collection_functions import *
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys