diff --git a/bob/learn/tensorflow/layers.py b/bob/learn/tensorflow/layers.py
index aa051071dd73f5dac54085d2f76931626fc32e69..6c2f4c06ea7a4ef2ae98f362146107dbf132d1d6 100644
--- a/bob/learn/tensorflow/layers.py
+++ b/bob/learn/tensorflow/layers.py
@@ -287,7 +287,7 @@ def add_bottleneck(model, bottleneck_size=128, dropout_rate=0.2):
 
     new_model.add(GlobalAvgPool2D())
     new_model.add(Dropout(dropout_rate, name="Dropout"))
-    new_model.add(Dense(128, use_bias=False, name="embeddings"))
+    new_model.add(Dense(bottleneck_size, use_bias=False, name="embeddings"))
     new_model.add(BatchNormalization(axis=-1, scale=False, name="embeddings/BatchNorm"))
 
     return new_model