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 4de599821..f93d1efd2 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
@@ -100,7 +100,9 @@ class RaggedTensorToTensorOpModel : public SingleOpModel {
   std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
 
   std::vector<float> GetOutputFloat() { return ExtractVector<float>(output_); }
-  std::vector<int32> GetOutputInt() { return ExtractVector<int32>(output_); }
+  std::vector<int32_t> GetOutputInt() {
+    return ExtractVector<int32_t>(output_);
+  }
 
   void InvokeFloat(const std::vector<int>& shape,
                    const std::vector<float>& values, float default_value,
@@ -114,7 +116,7 @@ class RaggedTensorToTensorOpModel : public SingleOpModel {
     SingleOpModel::Invoke();
   }
   void InvokeInt(const std::vector<int>& shape,
-                 const std::vector<int32>& values, int32 default_value,
+                 const std::vector<int32_t>& values, int32_t default_value,
                  const std::vector<std::vector<int>>& partition_values) {
     PopulateTensor(input_shape_, shape);
     PopulateTensor(input_values_, values);