Fix problem that test fail