another bug fix :/