Add support for Flow 0.8

Handle the Flow.Callback added by Flow 0.8