Fix some test case doesn' work