Fix ArgumentError: wrong number of arguments