another bug fix :(