diff --git a/Troubled_Cell_Detector.py b/Troubled_Cell_Detector.py
index 2d42b76db2997a0bb46f746a7ae6a6415c495725..a3d1002be90a21e97dcc1552b9d79800bf34025e 100644
--- a/Troubled_Cell_Detector.py
+++ b/Troubled_Cell_Detector.py
@@ -3,6 +3,7 @@
 @author: Laura C. Kühle, Soraya Terrab (sorayaterrab)
 
 TODO: Fix cell averages and reconstructions to create data with an x-point stencil
+TODO: Fix bug for ANN model initialization -> Done (later)
 
 """
 import os
@@ -207,7 +208,7 @@ class ArtificialNeuralNetwork(TroubledCellDetector):
 
         if not hasattr(ANN_Model, self._model):
             raise ValueError('Invalid model: "%s"' % self._model)
-        self._model = getattr(ANN_Model, self._model)(**self._model_config)
+        self._model = getattr(ANN_Model, self._model)(self._model_config)
 
     def get_cells(self, projection):
         num_ghost_cells = self._stencil_len//2