Fix bug regarding optional arguments