another bug fix :)