The id$ should be part of the entity instead of the args.
Merge pull request #2 from mcollina/custom-id2

The id$ should be part of the entity instead of the args.