fix a little bug.