Revert "fixed small bug"

This reverts commit 20d4861cf6baf653d24e06e9378779e3afabad96.