Fix bug introduced by last commit