Add extra argument check