Turn on optimisation for stage2 when validating