fix argument handling