Revert "minor bug"

This reverts commit f8d2633fd72279354093c02a0f499e55d807ef3c.