[imp] optimize code