Fix null value check for undrainable_node_behavior

Fix null value check for undrainable_node_behavior #834