diff --git a/tensorflow_privacy/privacy/optimizers/dp_optimizer.py b/tensorflow_privacy/privacy/optimizers/dp_optimizer.py
index 1d9c8cc..336c1d2 100644
--- a/tensorflow_privacy/privacy/optimizers/dp_optimizer.py
+++ b/tensorflow_privacy/privacy/optimizers/dp_optimizer.py
@@ -267,8 +267,8 @@ def make_optimizer_class(cls):
           ' called. Which means that the training is not differentially '
           'private. It happens for example in Keras training in TensorFlow '
           '2.0+.')
-      return super(DPOptimizerClass,
-                   self).apply_gradients(grads_and_vars, global_step, name)
+      return super(DPOptimizerClass, self).apply_gradients(
+          grads_and_vars=grads_and_vars, global_step=global_step, name=name)
 
   return DPOptimizerClass