Code optimisation

Small code optimisation.