fix argument bug