Fix "wrong number of arguments" error