fix the bug in the previous commit