Fixed bug introduced in previous commit