Callback fix

Forgot to upload the fix!