Getting the right callback.

Just a small fix for #3