Fix a bug in my last commit