Fixed to test to make it pass