small refine

small refine