Index: /issm/trunk/src/c/parallel/diagnostic_core_nonlinear.cpp
===================================================================
--- /issm/trunk/src/c/parallel/diagnostic_core_nonlinear.cpp	(revision 550)
+++ /issm/trunk/src/c/parallel/diagnostic_core_nonlinear.cpp	(revision 551)
@@ -59,6 +59,6 @@
 	
 	/*Initialize ug and ug_old */
-	if (numberofdofspernode==3)dofs[2]=1;//only keep vz if running with more than 3 dofs per node
-	if (numberofdofspernode==4)dofs[2]=1;dofs[3]=1;//only keep vz and presure if running with more than 3 dofs per node
+	if (numberofdofspernode>=3)dofs[2]=1;//only keep vz if running with more than 3 dofs per node
+	if (numberofdofspernode==4)dofs[3]=1;//only keep vz and presure if running with more than 3 dofs per node
 	ug=inputs->Get("velocity",&dofs[0],4);
 	VecDuplicate(ug,&old_ug);