diff --git a/lisa/tests/kernel/staging/load_tracking.py b/lisa/tests/kernel/staging/load_tracking.py
index d713abe1e93e4977d32861e8bd10741470848fb3..ff905b5670aeba4f4698167a6d8e5b1c5666f626 100644
--- a/lisa/tests/kernel/staging/load_tracking.py
+++ b/lisa/tests/kernel/staging/load_tracking.py
@@ -24,6 +24,7 @@ from lisa.tests.kernel.scheduler.load_tracking import (
     UTIL_SCALE,
     LoadTrackingBase
 )
+from lisa.env import TestEnv, ArtifactPath
 
 class CPUMigrationBase(LoadTrackingBase):
     """
@@ -73,6 +74,12 @@ class CPUMigrationBase(LoadTrackingBase):
 
         self.phases_durations = [phase.duration_s
                                  for phase in self.reference_task.phases]
+    @classmethod
+    def from_testenv(cls, te:TestEnv, res_dir:ArtifactPath=None) -> 'CPUMigrationBase':
+        """
+        Factory method to create a bundle using a live target
+        """
+        super().from_testenv(te=te, res_dir=res_dir)
 
     @classmethod
     def check_from_testenv(cls, te):
diff --git a/lisa/tests/kernel/test_bundle.py b/lisa/tests/kernel/test_bundle.py
index b8f5e5d693a1168f03cd8fe60eac54493ecf6d5f..9ed6a04555514950f9cbce8af482d5abcbde53de 100644
--- a/lisa/tests/kernel/test_bundle.py
+++ b/lisa/tests/kernel/test_bundle.py
@@ -287,7 +287,7 @@ class TestBundle(Serializable, abc.ABC):
             return False
 
     @classmethod
-    def from_testenv(cls, te:TestEnv, res_dir:ArtifactPath=None, **kwargs) -> 'TestBundle':
+    def from_testenv(cls, te, res_dir=None, **kwargs):
         """
         Factory method to create a bundle using a live target