Fix callback
Merge pull request #1 from toe/feature-fix-callback

Fix callback