Small fix for arguments