diff --git a/src/auto_scheduler/feature.cc b/src/auto_scheduler/feature.cc
index a2626ddba225..f8d7424d9d75 100644
--- a/src/auto_scheduler/feature.cc
+++ b/src/auto_scheduler/feature.cc
@@ -1222,7 +1222,7 @@ void GetPerStoreFeaturesWorkerFunc(const SearchTask& task, const State& state, i
     }
     auto mod = IRModule(Map<GlobalVar, BaseFunc>({{global_var, f}}));
 
-    if (task->target->id->device_type == kDLGPU) {
+    if (task->target->kind->device_type == kDLGPU) {
       auto pass_list = Array<tvm::transform::Pass>();
       // Phase 0
       pass_list.push_back(tir::transform::InjectPrefetch());