A little bug fix