diff --git a/python/tilus/hidet/include/tilus/tvm/ffi/extra_type_traits.h b/python/tilus/hidet/include/tilus/tvm/ffi/extra_type_traits.h
index 7569da75..a18ceb6b 100644
--- a/python/tilus/hidet/include/tilus/tvm/ffi/extra_type_traits.h
+++ b/python/tilus/hidet/include/tilus/tvm/ffi/extra_type_traits.h
@@ -16,6 +16,7 @@
 // helper used to produce human-readable error messages on type mismatch.
 #pragma once
 
+#include <tvm/ffi/container/tensor.h>
 #include <tvm/ffi/type_traits.h>
 #include <cuda.h>
 #include <cuda_fp16.h>