fix a bug

This PR still open for the test