diff --git a/prob_laplace.m b/prob_laplace.m
index bb79134e2ecad201745546880b9ca79e3477b160..636960acaec166e6c251a1f26d4500ed704b12dc 100644
--- a/prob_laplace.m
+++ b/prob_laplace.m
@@ -40,9 +40,11 @@ axis([0, 4, 0, 1.5])
 %%
 T_t = T1;
 ti=4;
-g01 = Ternary_model.gamma0(T_t.x(657), T_t.a, T_t.b, T_t.e_g0, T_t.u_g0, 0);
+g01 = Ternary_model.gamma0(T_t.x(657), T_t.a, T_t.b, T_t.e_g0, T_t.u_g0,...
+                           0, T_t.e, T_t.u0);
 pt1 = Ternary_model.phi_tot(T_t.x(656), T_t.a, T_t.b, T_t.e, T_t.u0, 0);
-g02 = Ternary_model.gamma0(T_t.x(657), T_t.a, T_t.b, T_t.e_g0, T_t.u_g0, 0);
+g02 = Ternary_model.gamma0(T_t.x(657), T_t.a, T_t.b, T_t.e_g0, T_t.u_g0,...
+                           0, T_t.e, T_t.u0);
 pt2 = Ternary_model.phi_tot(T_t.x(656), T_t.a, T_t.b, T_t.e, T_t.u0, 0);
 
 g0 = (g01+g02)/2;
@@ -190,7 +192,6 @@ end
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 params = {-5, b(7.7/3, 10^-6), 0.5, e(7.7/3), 0, 1, 10, 7, 0,...
           'Constituent', 0};
-%%
 t = [0, 0.05, 0.1, 1];
 direc = 1;
 x0 = sort(5-direc*(0:0.001:4.01));
@@ -366,7 +367,8 @@ pt = Ternary_model.phi_tot(x_interp, T_mov.a, T_mov.b, T_mov.e,...
 gra_a = Ternary_model.gradient_analytical(x_interp, T_mov.a, T_mov.b,...
                                           T_mov.e, T_mov.v*T_mov.t(i));
 g0 = Ternary_model.gamma0(x_interp, T_mov.a+T_mov.v*T_mov.t(i), T_mov.b,...
-                          T_mov.e_g0, T_mov.u_g0, T_mov.v*T_mov.t(i));
+                          T_mov.e_g0, T_mov.u_g0, T_mov.v*T_mov.t(i),...
+                          T_mov.e, T_mov.u0);
 dudx = diff(u)./diff(T_mov.x);
 f = -g0.*(1-pt)./pt.*(pt.*dudx-u_interp.*gra_a);
 s_dot(i) = sum(diff(T_mov.x).*f.^2./(g0.*(1-pt).*u_interp));