Fix the test bug, thanks for Ning and Izy