Fix test failed after 0ea83ff8478d867ebf1603a43cd5d3432022cee7