Revert "bug fix"

This reverts commit 9614ab54324c951cb6f86c224c34cd78d17212a7.