diff --git a/tests/lbpm_color_simulator.cpp b/tests/lbpm_color_simulator.cpp
index 18ac9a80..937c609a 100644
--- a/tests/lbpm_color_simulator.cpp
+++ b/tests/lbpm_color_simulator.cpp
@@ -39,6 +39,12 @@ int main(int argc, char **argv)
 		printf("Running Color LBM	\n");
 		printf("********************************************************\n");
 	}
+	// Initialize compute device
+	int device=ScaLBL_SetDevice(rank);
+	printf("Using GPU ID %i for rank %i \n",device,rank);
+	ScaLBL_DeviceBarrier();
+	MPI_Barrier(comm);
+	
     PROFILE_ENABLE(1);
     //PROFILE_ENABLE_TRACE();
     //PROFILE_ENABLE_MEMORY();