diff --git a/csrc/batch_decode.cu b/csrc/batch_decode.cu
index 7ffa762e87..de5e859662 100644
--- a/csrc/batch_decode.cu
+++ b/csrc/batch_decode.cu
@@ -13,6 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+#include <flashinfer/attention/decode.cuh>
 #include <flashinfer/attention/scheduler.cuh>
 #include <flashinfer/pos_enc.cuh>
 #include <flashinfer/utils.cuh>