diff --git a/tests/lbpm_color_macro_simulator.cpp b/tests/lbpm_color_macro_simulator.cpp
index 0c3a62b1..72013ca9 100644
--- a/tests/lbpm_color_macro_simulator.cpp
+++ b/tests/lbpm_color_macro_simulator.cpp
@@ -50,7 +50,7 @@ int main(int argc, char **argv)
 		}
 		// Initialize compute device
 		int device=ScaLBL_SetDevice(rank);
-		printf("Using GPU ID %i for rank %i \n",device,rank);
+		//printf("Using GPU ID %i for rank %i \n",device,rank);
 		ScaLBL_DeviceBarrier();
 		MPI_Barrier(comm);
 
diff --git a/tests/lbpm_segmented_decomp.cpp b/tests/lbpm_segmented_decomp.cpp
index f46f8360..05770f25 100644
--- a/tests/lbpm_segmented_decomp.cpp
+++ b/tests/lbpm_segmented_decomp.cpp
@@ -248,6 +248,8 @@ int main(int argc, char **argv)
 		MPI_Barrier(MPI_COMM_WORLD);
 		MPI_Bcast(LabelList,2*NLABELS,MPI_INT,0,MPI_COMM_WORLD);
 		
+		char *newIDs;
+		newIDs= new char [nx*ny*nz];
 		for (k=0;k<nz;k++){
 			for (j=0;j<ny;j++){
 				for (i=0;i<nx;i++){
@@ -255,7 +257,7 @@ int main(int argc, char **argv)
 					for (int label=0; label<NLABELS; label++){
 						int oldlabel=LabelList[2*label];
 						int newlabel=LabelList[2*label+1];
-						if (Dm.id[n]==char(oldlabel))  Dm.id[n] = char(newlabel);
+						if (Dm.id[n]==char(oldlabel))  newIDs[n] = char(newlabel);
 					}
 				}
 			}
@@ -271,6 +273,7 @@ int main(int argc, char **argv)
 				for (i=1;i<nx-1;i++){
 					n=k*nx*ny+j*nx+i;
 					total++;
+					Dm.id[n] = newIDs[n];
 					if (Dm.id[n] == 0){
 						count++;
 					}