fix little bug;