A quick bug fix