Fix for oil spill dataset demo