diff --git a/bayesflow/experimental/networks/coupling_flow/coupling_flow.py b/bayesflow/experimental/networks/coupling_flow/coupling_flow.py
index 5957ebf0..e4192455 100644
--- a/bayesflow/experimental/networks/coupling_flow/coupling_flow.py
+++ b/bayesflow/experimental/networks/coupling_flow/coupling_flow.py
@@ -39,7 +39,7 @@ def __init__(self, couplings: Sequence[AllInOneCoupling], base_distribution: Dis
     def all_in_one(
             cls,
             target_dim: int,
-            num_layers: 6,
+            num_layers=6,
             subnet_builder="default",
             transform="affine",
             permutation="fixed",