Fix bug introduced in previous commit