Fix to make IntegrationTestSimpleFlow work.