Bug fix, optimisation