Revert "bug fix"

This reverts commit 5d463f349b2ab5c0d07198908dda247931eab573.