Fixed a small bug introduced in the previous commit