Fix the bug introduced in the previous commit