diff --git a/tests/TestColorGradDFH.cpp b/tests/TestColorGradDFH.cpp
index b659f3bf..59cc7a5d 100644
--- a/tests/TestColorGradDFH.cpp
+++ b/tests/TestColorGradDFH.cpp
@@ -193,7 +193,6 @@ int main(int argc, char **argv)
 			}
 		}
 		int Npad=(Np/16 + 2)*16;
-
 		int *neighborList;
 		IntArray Map(Nx,Ny,Nz);
 		neighborList= new int[18*Npad];
@@ -204,7 +203,6 @@ int main(int argc, char **argv)
 		int dist_mem_size = Np*sizeof(double);
 		int neighborSize=18*Np*sizeof(int);
 		if (rank==0)	printf ("Allocating distributions \n");
-
 		int *NeighborList;
 		int *dvcMap;
 		double *Phi;
@@ -254,8 +252,12 @@ int main(int argc, char **argv)
 		ScaLBL_CopyToDevice(Phi, PHASE, Np*sizeof(double));
 		//...........................................................................
 
-		ScaLBL_D3Q19_Gradient_DFH(NeighborList, Phi, ColorGrad, Potential, 0, Np, Np);
-	
+		// compute the gradient 
+		ScaLBL_D3Q19_Gradient_DFH(neighborList, Phi, ColorGrad, SolidPotential, ScaLBL_Comm.first_interior, ScaLBL_Comm.last_interior, Np);
+		ScaLBL_Comm.SendHalo(Phi);
+		ScaLBL_D3Q19_Gradient_DFH(neighborList, Phi, ColorGrad, SolidPotential, 0, ScaLBL_Comm.first_interior, Np);
+		ScaLBL_Comm.RecvGrad(ColorGrad);
+		
     	double *COLORGRAD;
     	COLORGRAD= new double [3*Np];
     	int SIZE=3*Np*sizeof(double);
diff --git a/tests/lbpm_dfh_simulator.cpp b/tests/lbpm_dfh_simulator.cpp
index 513b3af0..63e34c62 100644
--- a/tests/lbpm_dfh_simulator.cpp
+++ b/tests/lbpm_dfh_simulator.cpp
@@ -509,6 +509,7 @@ int main(int argc, char **argv)
 		double *Tmp;
 		Tmp=new double[3*Np];
 		Averages->UpdateMeshValues(); // this computes the gradient of distance field (among other things)
+		double count_wet=0.f;
 		for (k=1; k<Nz-1; k++){
 			for (j=1; j<Ny-1; j++){
 				for (i=1; i<Nx-1; i++){
@@ -524,8 +525,10 @@ int main(int argc, char **argv)
 						Tmp[idx+Np] = value*dy;
 						Tmp[idx+2*Np] = value*dz;
 						// initialize fluid phases
-						if (Dm.id[n] == 1)	PhaseLabel[idx] = 1.0;
-						else 				PhaseLabel[idx] = -1.0;
+						if (Mask.id[n] == 1)	PhaseLabel[idx] = 1.0;
+						else {
+							PhaseLabel[idx] = -1.0;
+						}
 					}
 				}
 			}