diff --git a/kernels/portable/cpu/util/targets.bzl b/kernels/portable/cpu/util/targets.bzl
index cdd3695cec7..d942f7ad9f9 100644
--- a/kernels/portable/cpu/util/targets.bzl
+++ b/kernels/portable/cpu/util/targets.bzl
@@ -273,6 +273,7 @@ def define_common_targets():
         srcs = ["tensor_util.cpp"],
         exported_headers = ["tensor_util.h"],
         deps = [
+            "//executorch/runtime/core/exec_aten:lib",
             "//executorch/runtime/kernel:kernel_includes",
         ],
         visibility = ["//executorch/kernels/portable/cpu/..."],
diff --git a/kernels/portable/cpu/util/tensor_util.cpp b/kernels/portable/cpu/util/tensor_util.cpp
index 51d60f82d81..abdb202dfb6 100644
--- a/kernels/portable/cpu/util/tensor_util.cpp
+++ b/kernels/portable/cpu/util/tensor_util.cpp
@@ -1,5 +1,3 @@
-#include "tensor_util.h"
-
 /*
  * Copyright (c) Meta Platforms, Inc. and affiliates.
  * All rights reserved.
@@ -8,8 +6,8 @@
  * LICENSE file in the root directory of this source tree.
  */
 
+#include <executorch/kernels/portable/cpu/util/tensor_util.h>
 #include <executorch/runtime/core/exec_aten/util/tensor_util.h>
-
 #include <executorch/runtime/platform/assert.h>
 
 namespace executorch::runtime {