Fix bug introduced in last commit