fix bug
refine