diff --git a/taichi/backends/cuda/jit_cuda.cpp b/taichi/backends/cuda/jit_cuda.cpp
index 71ddd99cc51f9..f80d89b47424e 100644
--- a/taichi/backends/cuda/jit_cuda.cpp
+++ b/taichi/backends/cuda/jit_cuda.cpp
@@ -18,10 +18,6 @@
 #include "llvm/Target/TargetMachine.h"
 #include "llvm/ExecutionEngine/Orc/JITTargetMachineBuilder.h"
 
-#define TI_RUNTIME_HOST
-#include "taichi/program/context.h"
-#undef TI_RUNTIME_HOST
-
 #include "taichi/backends/cuda/cuda_context.h"
 #include "taichi/backends/cuda/cuda_driver.h"
 #include "taichi/jit/jit_session.h"
@@ -30,6 +26,10 @@
 #include "taichi/system/timer.h"
 #include "taichi/util/file_sequence_writer.h"
 
+#define TI_RUNTIME_HOST
+#include "taichi/program/context.h"
+#undef TI_RUNTIME_HOST
+
 TLANG_NAMESPACE_BEGIN
 
 #if defined(TI_WITH_CUDA)