[Flow] Register extension

Change-Id: I5022d972f13b4dcafc95870b0ecec494df238dc6