slight bug fix again.