Another silly bug fix