Quick silly bug fix