Small bit of optimisation