Fix test to make it pass