Fix bug for optional arguments