diff --git a/experiment.py b/experiment.py
index 511c989..8816b05 100644
--- a/experiment.py
+++ b/experiment.py
@@ -35,9 +35,8 @@ def train_with_parameters(
     """
     model = constellation.ConstellationNet(
         order=order,
-        encoder_layers_sizes=layer_sizes,
-        decoder_layers_sizes=layer_sizes[::-1],
-        channel_model=constellation.GaussianChannel()
+        encoder_layers=layer_sizes,
+        decoder_layers=layer_sizes[::-1],
     )
 
     # List of training examples (not shuffled)