Revert "Fix possible bug"

This reverts commit e06d095fa119c9eb4b79850625a24c147bd509f0.