Fixing watershed image output