Fix flow plotting