Fix to the input scaling testing