A little bug fixed