improve cmd code
minor change