quick bug fix