small  bug fix