Revert "fixed bug"

This reverts commit 781ee4809ee62336b6be538c3d9e4d67ba7dbb81.
Updated to revert