Small code optimisation