Changed testing in setup