fixed the bug introduced in the last commit