Add a quick way of overriding which flow to use