Update StateNode.cpp