bug fix (hopefully)