Input argument fix