Correctly flag nodes