diff --git a/Mean_Drift/WECSim/userDefinedFunctions.m b/Mean_Drift/WECSim/userDefinedFunctions.m
index 2cd6f9a0..d3c434a9 100644
--- a/Mean_Drift/WECSim/userDefinedFunctions.m
+++ b/Mean_Drift/WECSim/userDefinedFunctions.m
@@ -1,12 +1,12 @@
 inds = find(output.wave.time >= simu.rampTime + waves.period*5);
 %%
-surgeMeanDriftTH = body.hydroForce.fExt.md(1)*waves.A^2;
+surgeMeanDriftTH = body.hydroForce.fExt.md(1)*waves.amplitude^2;
 surgeMeanDriftWS = mean(output.bodies.forceExcitation(inds,1));
 %%
 close all
-plot(output.bodies.time/waves.period,output.bodies.forceExcitation(:,1)/(simu.rho*simu.g*waves.A),...
-     output.bodies.time/waves.period,ones(size(output.bodies.time)).*mean(output.bodies.forceExcitation(inds,1)/(simu.rho*simu.g*waves.A)),'--',...
-     output.bodies.time/waves.period,ones(size(output.bodies.time)).*surgeMeanDriftTH/(simu.rho*simu.g*waves.A),'-.');grid on
+plot(output.bodies.time/waves.period,output.bodies.forceExcitation(:,1)/(simu.rho*simu.gravity*waves.amplitude),...
+     output.bodies.time/waves.period,ones(size(output.bodies.time)).*mean(output.bodies.forceExcitation(inds,1)/(simu.rho*simu.gravity*waves.amplitude)),'--',...
+     output.bodies.time/waves.period,ones(size(output.bodies.time)).*surgeMeanDriftTH/(simu.rho*simu.gravity*waves.amplitude),'-.');grid on
 xlabel('Time, t/T, [-]');ylabel('Surge Wave Excitation Force, f_{ext,1}/\rhogA, [N/(N/m^{2})]');
 legend('Surge Excitation Force','Surge Mean Drift Force WEC-Sim','Surge Mean Drift Force Theory')
 figure()
diff --git a/Mean_Drift/WECSim/wecSimInputFile.m b/Mean_Drift/WECSim/wecSimInputFile.m
index ef2ccfb7..b42b0037 100644
--- a/Mean_Drift/WECSim/wecSimInputFile.m
+++ b/Mean_Drift/WECSim/wecSimInputFile.m
@@ -30,6 +30,7 @@
 %      0 (no), 
 %      1 (from control surface) 
 %      2 (from momentum conservation)
+%      3 (from pressure integration)
 body(1).meanDrift = 1;                                
 
 %% PTO and Constraint Parameters