Fixing after_success hook.