Fix flow test setup