Fix some test case doesn' work

Fix some test case doesn' work