fix the little bug