a little bug fix.