diff --git a/tensorflow/compiler/mlir/xla/tests/legalize-tf-full-conversion.mlir b/tensorflow/compiler/mlir/xla/tests/legalize-tf-full-conversion.mlir
index 932b661433f..d2b4d269fef 100644
--- a/tensorflow/compiler/mlir/xla/tests/legalize-tf-full-conversion.mlir
+++ b/tensorflow/compiler/mlir/xla/tests/legalize-tf-full-conversion.mlir
@@ -1,4 +1,4 @@
-// RUN: tf-opt %s -xla-legalize-tf -verify-diagnostics
+// RUN: tf-opt %s -xla-legalize-tf -split-input-file -verify-diagnostics
 
 func @tf_executor_graph_op() {
     // expected-error@+1 {{failed to legalize operation 'tf_executor.graph'}}
@@ -13,12 +13,16 @@ func @tf_executor_graph_op() {
 
 }
 
+// -----
+
 func @tf_unknown_op(%arg0: tensor<2xi32>) -> tensor<2xi32> {
   // expected-error@+1 {{failed to legalize operation 'tf.OpA'}}
   %0 = "tf.OpA"(%arg0, %arg0) : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi32>
   return %0: tensor<2xi32>
 }
 
+// -----
+
 func @tf_known_op(%arg0: tensor<2xi32>) -> tensor<2xi32> {
   %0 = "tf.Add"(%arg0, %arg0) : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi32>
   return %0: tensor<2xi32>