diff --git a/src/architectures.py b/src/architectures.py
index a1e18a4..68645ea 100644
--- a/src/architectures.py
+++ b/src/architectures.py
@@ -136,7 +136,7 @@ def summary(self, line_length=None, positions=None, print_fn=None):
                         self.nr_bands), name='input')
         model = Model(inputs=[inputs], outputs=self.call(inputs),
                       name=self.name)
-        return model.summary()
+        return model.summary(line_length, positions, print_fn)
 
     @abstractmethod
     def instantiate_layers(self):