diff --git a/setup.py b/setup.py
index d219135992f..00cbe2e7bdf 100644
--- a/setup.py
+++ b/setup.py
@@ -599,7 +599,9 @@ def run(self):
         # https://discourse.cmake.org/t/installing-headers-the-modern-way-regurgitated-and-revisited/3238/3
         for include_dir in [
             "runtime/core/",
+            "runtime/executor/",
             "runtime/kernel/",
+            "runtime/backend/",
             "runtime/platform/",
             "extension/kernel_util/",
             "extension/tensor/",