Fixed a small bug introduced in last commit