Just a quick bug fix