diff --git a/tests/integration/states/host.py b/tests/integration/states/host.py
index 524f7a2c15..8fec3174ed 100644
--- a/tests/integration/states/host.py
+++ b/tests/integration/states/host.py
@@ -6,20 +6,12 @@ tests for host state
 import os
 import shutil
 
+# Import Salt Testing libs
+from salttesting.helpers import ensure_in_syspath
+ensure_in_syspath('../../')
+
 # Import salt libs
-try:
-    import integration
-except ImportError:
-    if __name__ == '__main__':
-        import sys
-        sys.path.insert(
-            0, os.path.abspath(
-                os.path.join(
-                    os.path.dirname(__file__), '../../'
-                )
-            )
-        )
-    import integration
+import integration
 import salt.utils
 
 HFILE = os.path.join(integration.TMP, 'hosts')