Fix an bug introduced in the previous commit