Both tests now should pass