Fix bug introduced by previous commit