Fixed error with wrong number of arguments