add little optimisation