Fix test to pass