Pass the wrapped entity to a callback method.
Merge pull request #166 from Gizra/callback-fix

Pass the wrapped entity to a callback method