Fix error in argument handling