updated the setup of the tests