Fix tiny bug in the latest commit