Bug fix: Change the way of getting argument.