Small optimisation fixes