diff --git a/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc b/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
index 5dc2b6d..489fa66 100644
--- a/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
+++ b/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
@@ -121,7 +121,6 @@
           if (dw_params.padding_type == PaddingType::kValid &&
               dw_params.stride_width == 1 && dw_params.stride_height == 1 &&
               filter_shape.Dims(1) == 1 && filter_shape.Dims(2) == 3 &&
-              dw_params.dilation_width_factor == 1 &&
               dw_params.dilation_height_factor == 1) {
             kelvin::opt::DepthwiseConv2DKelvinS16K3x1(
                 tflite::micro::GetTensorData<int16_t>(input),
@@ -129,6 +128,7 @@
                 tflite::micro::GetOptionalTensorData<int64_t>(bias),
                 filter_shape.Dims(3) /*channels*/,
                 input_shape.Dims(2) /*frames*/,
+                dw_params.dilation_width_factor /*dilation*/,
                 data.per_channel_output_multiplier,
                 data.per_channel_output_shift,
                 dw_params.quantized_activation_min,