diff --git a/riid/models/neural_nets/basic.py b/riid/models/neural_nets/basic.py
index 48053d6..705b32e 100644
--- a/riid/models/neural_nets/basic.py
+++ b/riid/models/neural_nets/basic.py
@@ -129,7 +129,7 @@ def fit(self, ss: SampleSet, batch_size: int = 200, epochs: int = 20,
                 kernel_regularizer=l2(self.l2_alpha),
             )(inputs)
             if self.dropout is not None:
-                last_layer = Dropout(0.2)(dense_layer)
+                last_layer = Dropout(self.dropout)(dense_layer)
             else:
                 last_layer = dense_layer
             outputs = Dense(Y.shape[1], activation=self.final_activation)(last_layer)