Use flower.common.Callback in a few places