fixed a bug of not using provided args.K