Revert "Fix bug"

This reverts commit a7d83c480729c1f74a1c5704f1087bd19f06cf05.