little bug fixed