fixixng test error in 0.6