Fix another minor bug in the previous commit