Bug fix
Minor optimisation