code refine
fix bug