Fixed bug pointed out by Chaitanya in sponge_flux