Little modify

Little modify