minor bug fix again