diff --git a/src/python/subclassing.py b/src/python/subclassing.py
index fe2674ec..bccbef29 100644
--- a/src/python/subclassing.py
+++ b/src/python/subclassing.py
@@ -65,7 +65,7 @@ class ConvNet(Model):
         x = self.maxpool2(x)
         x = self.flatten(x)
         x = self.fc1(x)
-        x = self.dropout(x, training=is_training)
+        x = self.dropout(x)
         x = self.out(x)
         if not is_training:
             # tf cross entropy expect logits without softmax, so only