diff --git a/tensorflow_text/core/kernels/ragged_tensor_to_tensor_tflite_test.cc b/tensorflow_text/core/kernels/ragged_tensor_to_tensor_tflite_test.cc
index e1241084d..347f47126 100644
--- a/tensorflow_text/core/kernels/ragged_tensor_to_tensor_tflite_test.cc
+++ b/tensorflow_text/core/kernels/ragged_tensor_to_tensor_tflite_test.cc
@@ -93,8 +93,7 @@ class RaggedTensorToTensorOpModel : public SingleOpModel {
     BuildInterpreter(shapes, /*num_threads=*/-1,
                      /*allow_fp32_relax_to_fp16=*/false,
                      /*apply_delegate=*/true,
-                     /*allocate_and_delegate=*/allocate_and_delegate,
-                     /*use_simple_allocator=*/false);
+                     /*allocate_and_delegate=*/allocate_and_delegate);
   }
 
   std::vector<int> GetOutputShape() { return GetTensorShape(output_); }