Fix test mishap