diff --git a/research/object_detection/core/freezable_sync_batch_norm.py b/research/object_detection/core/freezable_sync_batch_norm.py
index f95a1064983..a7753da4994 100644
--- a/research/object_detection/core/freezable_sync_batch_norm.py
+++ b/research/object_detection/core/freezable_sync_batch_norm.py
@@ -17,8 +17,7 @@
 import tensorflow as tf
 
 
-class FreezableSyncBatchNorm(tf.keras.layers.experimental.SyncBatchNormalization
-                            ):
+class FreezableSyncBatchNorm(tf.keras.layers.BatchNormalization):
   """Sync Batch normalization layer (Ioffe and Szegedy, 2014).
 
   This is a `freezable` batch norm layer that supports setting the `training`