fix control bug