fix the important bug