fix bug

This reverts commit 960f1a1a88e5a73596907adfb85694227a401723.