Add check for node assignment