slight optimisation

^