diff --git a/test/test_data_handler/test_iterator.py b/test/test_data_handler/test_iterator.py
index e47d725a4fd78fec98e81a6de9c18869e7b47637..bb8ecb5d216519b3662a5baa4d463780b4c29d8c 100644
--- a/test/test_data_handler/test_iterator.py
+++ b/test/test_data_handler/test_iterator.py
@@ -106,6 +106,9 @@ class DummyData:
         Y2 = np.random.randint(21, 30, size=(self.number_of_samples, 5, 1))  # samples, window, variables
         return [Y1, Y2]
 
+    def get_data(self, upsampling=False, as_numpy=True):
+        return self.get_X(upsampling, as_numpy), self.get_Y(upsampling, as_numpy)
+
 
 class TestKerasIterator:
 
diff --git a/test/test_run_modules/test_model_setup.py b/test/test_run_modules/test_model_setup.py
index 962287e09aacd3c44961a827c86b331d643ec401..6e8d3ea9ebab40c79b17b2fba386322a630f00e1 100644
--- a/test/test_run_modules/test_model_setup.py
+++ b/test/test_run_modules/test_model_setup.py
@@ -150,3 +150,6 @@ class DummyData:
         Y1 = np.random.randint(0, 10, size=(self.number_of_samples, 5))  # samples, window
         Y2 = np.random.randint(21, 30, size=(self.number_of_samples, 3))  # samples, window
         return [Y1, Y2]
+
+    def get_data(self, upsampling=False, as_numpy=True):
+        return self.get_X(upsampling, as_numpy), self.get_Y(upsampling, as_numpy)