Minor bug fix!