Revert "bug"

This reverts commit ceda61c92b9b01bf612efde2f932ff9f738a4544.