Revise a bug in the previous commit