I would look at your infusion rate and hydration. At the beginning it is more important that the rate is very slow. And super hydration the day before, day of and day after is also very important. If you don't have any sodium issues, eating salty snacks will keep more water in your system.