diff --git a/test/integration/test_kubernetes_staging.py b/test/integration/test_kubernetes_staging.py
index 5e60c99a1f7..25994b8a0d0 100644
--- a/test/integration/test_kubernetes_staging.py
+++ b/test/integration/test_kubernetes_staging.py
@@ -25,6 +25,7 @@ from galaxy_test.driver import integration_util
 from .test_local_job_cancellation import CancelsJob
 from .test_containerized_jobs import EXTENDED_TIMEOUT, MulledJobTestCases
 from .test_job_environments import BaseJobEnvironmentIntegrationTestCase
+from .test_kubernetes_runner import KubernetesDatasetPopulator
 from .test_local_job_cancellation import CancelsJob
 
 TOOL_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, 'tools'))
@@ -120,6 +121,7 @@ class BaseKubernetesStagingTest(BaseJobEnvironmentIntegrationTestCase, MulledJob
 
     def setUp(self):
         super(BaseKubernetesStagingTest, self).setUp()
+        self.dataset_populator = KubernetesDatasetPopulator(self.galaxy_interactor)
         self.history_id = self.dataset_populator.new_history()
 
     @classmethod