diff --git a/tensorflow/compiler/mlir/lite/experimental/tac/utils/utils.h b/tensorflow/compiler/mlir/lite/experimental/tac/utils/utils.h
index 049cc18687a669..eecf8bd7467f57 100644
--- a/tensorflow/compiler/mlir/lite/experimental/tac/utils/utils.h
+++ b/tensorflow/compiler/mlir/lite/experimental/tac/utils/utils.h
@@ -16,6 +16,7 @@ limitations under the License.
 #ifndef TENSORFLOW_COMPILER_MLIR_LITE_EXPERIMENTAL_TAC_UTILS_UTILS_H_
 #define TENSORFLOW_COMPILER_MLIR_LITE_EXPERIMENTAL_TAC_UTILS_UTILS_H_
 
+#include <optional>
 #include <string>
 
 #include "absl/status/status.h"