Fixed bug

Fixed bug
Revert "Fixed bug"

This reverts commit 11da49620ff6b2cfab920ec1cff7b1d8c7097cd1.