Revert "fix bug"

This reverts commit f9ad1196c64c95e4a502846cb01826205dc7eef5.
fix bug again