a little bug fix