diff --git a/tensorflow_privacy/privacy/optimizers/dp_optimizer_keras_vectorized.py b/tensorflow_privacy/privacy/optimizers/dp_optimizer_keras_vectorized.py
index 16a12d5..3915682 100644
--- a/tensorflow_privacy/privacy/optimizers/dp_optimizer_keras_vectorized.py
+++ b/tensorflow_privacy/privacy/optimizers/dp_optimizer_keras_vectorized.py
@@ -252,8 +252,8 @@ def make_vectorized_keras_optimizer_class(cls):
 
 
 VectorizedDPKerasAdagradOptimizer = make_vectorized_keras_optimizer_class(
-    tf.keras.optimizers.Adagrad)
+    tf.keras.optimizers.legacy.Adagrad)
 VectorizedDPKerasAdamOptimizer = make_vectorized_keras_optimizer_class(
-    tf.keras.optimizers.Adam)
+    tf.keras.optimizers.legacy.Adam)
 VectorizedDPKerasSGDOptimizer = make_vectorized_keras_optimizer_class(
-    tf.keras.optimizers.SGD)
+    tf.keras.optimizers.legacy.SGD)