fixed a little bug.