fix argument stuff for now