fix a important bug