diff --git a/include/flashinfer/comm/trtllm_allreduce_fusion.cuh b/include/flashinfer/comm/trtllm_allreduce_fusion.cuh
index 8ec84c4a03..142da62c5c 100644
--- a/include/flashinfer/comm/trtllm_allreduce_fusion.cuh
+++ b/include/flashinfer/comm/trtllm_allreduce_fusion.cuh
@@ -8,6 +8,7 @@
 #endif
 
 #include <cuda/std/optional>
+#include <optional>
 #include <tuple>
 #include <type_traits>
 
diff --git a/include/flashinfer/comm/trtllm_moe_allreduce_fusion.cuh b/include/flashinfer/comm/trtllm_moe_allreduce_fusion.cuh
index 143e25de9c..e11589dda3 100644
--- a/include/flashinfer/comm/trtllm_moe_allreduce_fusion.cuh
+++ b/include/flashinfer/comm/trtllm_moe_allreduce_fusion.cuh
@@ -8,6 +8,7 @@
 #endif
 
 #include <cuda/std/optional>
+#include <optional>
 #include <tuple>
 #include <type_traits>
 
diff --git a/version.txt b/version.txt
index 267577d47e..abaf45803d 100644
--- a/version.txt
+++ b/version.txt
@@ -1 +1 @@
-0.4.1
+0.4.1.2