Optimisation fixes