Fixed bug introduced in previous commit :S