[Fix] fix bug of arguments