fixed bug fix :)