fix handling argument
fix handling argument