Revert "Fix bug"

This reverts commit 2c5404e3dc01077a44bcbb5875431b766593648e.