Fixing a bug - Wrong number of arguments