Free Energy Integration bug fix