Small code simplification

If I understand the intention correctly.