Revert "remove bug"

This reverts commit a49b3dbf080ec065e7fafbdd83f2fba7be775d6f.