diff --git a/tajo-core/src/main/java/org/apache/tajo/querymaster/Stage.java b/tajo-core/src/main/java/org/apache/tajo/querymaster/Stage.java
index 20add9fa9b..5a38573cbb 100644
--- a/tajo-core/src/main/java/org/apache/tajo/querymaster/Stage.java
+++ b/tajo-core/src/main/java/org/apache/tajo/querymaster/Stage.java
@@ -468,8 +468,12 @@ private StageHistory makeStageHistory() {
     stageHistory.setKilledObjectCount(killedObjectCount);
     stageHistory.setFailedObjectCount(failedObjectCount);
     stageHistory.setTotalScheduledObjectsCount(totalScheduledObjectsCount);
-    stageHistory.setHostLocalAssigned(getTaskScheduler().getHostLocalAssigned());
-    stageHistory.setRackLocalAssigned(getTaskScheduler().getRackLocalAssigned());
+
+    AbstractTaskScheduler scheduler = getTaskScheduler();
+    if (scheduler != null) {
+      stageHistory.setHostLocalAssigned(scheduler.getHostLocalAssigned());
+      stageHistory.setRackLocalAssigned(scheduler.getRackLocalAssigned());
+    }
 
     long totalInputBytes = 0;
     long totalReadBytes = 0;