diff --git a/mlp/models.py b/mlp/models.py
index 63d034f..7f1273e 100644
--- a/mlp/models.py
+++ b/mlp/models.py
@@ -80,7 +80,7 @@ class MultipleLayerModel(object):
         """A list of all of the parameters of the model."""
         params = []
         for layer in self.layers:
-            if isinstance(layer, LayerWithParameters):
+            if isinstance(layer, LayerWithParameters) or isinstance(layer, StochasticLayerWithParameters):
                 params += layer.params
         return params