Fix test bug.