Fix a tiny bug of the last commit