code refine
minor change