Fix test so it passes