made changes to the tests