Fix another minor bug