diff --git a/compiler/plugins/input/Torch/torch-iree/InputConversion/test/scatter.mlir b/compiler/plugins/input/Torch/torch-iree/InputConversion/test/scatter.mlir
index 0609510..e3debb5 100644
--- a/compiler/plugins/input/Torch/torch-iree/InputConversion/test/scatter.mlir
+++ b/compiler/plugins/input/Torch/torch-iree/InputConversion/test/scatter.mlir
@@ -3,7 +3,7 @@
 func.func @scatter_update(
     %original: tensor<8xi32>, %indices: tensor<3x1xi32>,
     %updates: tensor<3xi32>) -> tensor<8xi32> {
-  %0 = tm_tensor.scatter unique_indices(true)
+  %0 = tm_tensor.scatter {dimension_map = array<i64: 0>} unique_indices(true)
     ins(%updates, %indices : tensor<3xi32>, tensor<3x1xi32>)
     outs(%original : tensor<8xi32>)  {
   ^bb0(%update: i32, %orig: i32):  // no predecessors
@@ -29,7 +29,7 @@
 func.func @scatter_add(
     %original: tensor<8xi32>, %indices: tensor<3x1xi32>,
     %updates: tensor<3xi32>) -> tensor<8xi32> {
-  %0 = tm_tensor.scatter unique_indices(true)
+  %0 = tm_tensor.scatter {dimension_map = array<i64: 0>} unique_indices(true)
     ins(%updates, %indices : tensor<3xi32>, tensor<3x1xi32>)
     outs(%original : tensor<8xi32>)  {
   ^bb0(%update: i32, %orig: i32):  // no predecessors
diff --git a/third_party/torch-mlir b/third_party/torch-mlir
index 0f80e75..e48fe45 160000
--- a/third_party/torch-mlir
+++ b/third_party/torch-mlir
@@ -1 +1 @@
-Subproject commit 0f80e75c2eb6dfed00bf051644a5e3fb97207bb8
+Subproject commit e48fe4588631e7a37a2899f9d4cd5c4cbc967481