Updated the tests file