Correct bug from previous commit