Fixed small bug introduced in previous commit