[fix]:improve code