Fix bug in argument