Correct bug in previous commit