bug fix :