fix bug to pass arguments