Fix argument length bug