diff --git a/benchmarks/DeepLearning/Models/MobileNet-V2/MobileNet.mlir b/benchmarks/DeepLearning/Models/MobileNet-V2/MobileNet.mlir
index f2832087d7786861fdcdfaeb969853cac60801a7..4187c5585297ea9f91f5ddee3b949783d6dbbcbc 100644
--- a/benchmarks/DeepLearning/Models/MobileNet-V2/MobileNet.mlir
+++ b/benchmarks/DeepLearning/Models/MobileNet-V2/MobileNet.mlir
@@ -3783,7 +3783,7 @@ module  {
       %1328 = arith.addf %arg1, %arg2 : f32
       linalg.yield %1328 : f32
     } -> tensor<1x1x1x1001xf32>
-    %1327 = linalg.tensor_collapse_shape %1326 [[0], [1, 2, 3]] : tensor<1x1x1x1001xf32> into tensor<1x1001xf32>
+    %1327 = tensor.collapse_shape %1326 [[0], [1, 2, 3]] : tensor<1x1x1x1001xf32> into tensor<1x1001xf32>
     return %1327 : tensor<1x1001xf32>
   }
 }