fixed small bug introduced by previous commit