fix handling argument