quick bug fix ;p