Fix a bug introduced in last commit