Fix test bug for py26