Fixing after adding image normalisation cos I broke it din i