fix: correct bug introduced in the last commit