diff --git a/ge/hybrid/executor/node_state.h b/ge/hybrid/executor/node_state.h
index ef348a69..84a52abd 100644
--- a/ge/hybrid/executor/node_state.h
+++ b/ge/hybrid/executor/node_state.h
@@ -30,7 +30,7 @@ class NodeTask;
 struct GraphExecutionContext;
 class SubgraphContext;
 class TaskContext;
-class NodeState;
+struct NodeState;
 
 class ShapeFuture {
  public: