Fix test

Fix some failed unit test