Fix a bug introduced in my last commit