Revert "bug fix"

This reverts commit 3dcffad337ed994cb58a642d64bd517a17458b54.