Fixing the tests file