Fixed the input/output args.