Fix to parse arguments