Fix some test error