diff --git a/tensorflow_model_optimization/python/core/clustering/keras/clustering_registry.py b/tensorflow_model_optimization/python/core/clustering/keras/clustering_registry.py
index d03a8cb45..f832e2bc0 100644
--- a/tensorflow_model_optimization/python/core/clustering/keras/clustering_registry.py
+++ b/tensorflow_model_optimization/python/core/clustering/keras/clustering_registry.py
@@ -155,7 +155,6 @@ class ClusteringLookupRegistry(object):
       layers.Conv2DTranspose: {'kernel': ConvolutionalWeightsCA},
       layers.Conv3D: {'kernel': ConvolutionalWeightsCA},
       layers.Conv3DTranspose: {'kernel': ConvolutionalWeightsCA},
-      layers.DepthwiseConv2D: {'depthwise_kernel': ConvolutionalWeightsCA},
       layers.SeparableConv1D: {'pointwise_kernel': ConvolutionalWeightsCA},
       layers.SeparableConv2D: {'pointwise_kernel': ConvolutionalWeightsCA},
       layers.Dense: {'kernel': DenseWeightsCA},
@@ -244,7 +243,7 @@ class ClusteringRegistry(object):
       layers.Cropping1D: [],
       layers.Cropping2D: [],
       layers.Cropping3D: [],
-      layers.DepthwiseConv2D: ['depthwise_kernel'],
+      layers.DepthwiseConv2D: [],
       layers.SeparableConv1D: ['pointwise_kernel'],
       layers.SeparableConv2D: ['pointwise_kernel'],
       layers.UpSampling1D: [],