slight optimisation tweak