Fix flow tests