Fixed optimisation bug