diff --git a/src/data_handling/iterator.py b/src/data_handling/iterator.py
index e6831e44f53bb545f177fff23e63a05f78cbeb93..51073286810f4eb6ef02beb47727932008cb9b7c 100644
--- a/src/data_handling/iterator.py
+++ b/src/data_handling/iterator.py
@@ -130,7 +130,7 @@ class KerasIterator(keras.utils.Sequence):
             np.random.shuffle(self.indexes)
 
 
-class DummyData:
+class DummyData:  # pragma: no cover
 
     def __init__(self, number_of_samples=np.random.randint(100, 150)):
         self.number_of_samples = number_of_samples