Fixed a small bug introduced by the previous changes.