diff --git a/test/test_data_handler/test_iterator.py b/test/test_data_handler/test_iterator.py
index fe740c094b41df7da45bd0f76d678830f95e1902..b5fb30a90c99d33e9dfe3db1346cfd7f43549fc9 100644
--- a/test/test_data_handler/test_iterator.py
+++ b/test/test_data_handler/test_iterator.py
@@ -331,7 +331,6 @@ class TestPermuteData:
 
 class TestFProc:
 
-
     @pytest.fixture
     def collection(self):
         coll = []
diff --git a/test/test_run_modules/test_model_setup.py b/test/test_run_modules/test_model_setup.py
index 6e8d3ea9ebab40c79b17b2fba386322a630f00e1..83fe409fbf9aa5237338daf88b479630da0e81f2 100644
--- a/test/test_run_modules/test_model_setup.py
+++ b/test/test_run_modules/test_model_setup.py
@@ -140,6 +140,12 @@ class DummyData:
     def __init__(self, number_of_samples=np.random.randint(100, 150)):
         self.number_of_samples = number_of_samples
 
+    def __len__(self, upsampling=False):
+        if upsampling is False:
+            return self._len
+        else:
+            return self._len_upsampling
+
     def get_X(self, upsampling=False, as_numpy=True):
         X1 = np.random.randint(0, 10, size=(self.number_of_samples, 14, 1, 5))  # samples, window, variables
         X2 = np.random.randint(21, 30, size=(self.number_of_samples, 10, 1, 2))  # samples, window, variables