A little bug fixed #4