Fixed visualization error for returned flowspace