Fix a bug in the last commit