slight bug fix