diff --git a/test/rt/nodes/taskiter.c b/test/rt/nodes/taskiter.c
index 7d4aead..0ab32c8 100644
--- a/test/rt/nodes/taskiter.c
+++ b/test/rt/nodes/taskiter.c
@@ -10,17 +10,20 @@ int
 main(void)
 {
 	int A;
-	#pragma oss task out(A)
+	#pragma oss task out(A) label("init")
 	A = 1;
 
-	#pragma oss taskiter in(A) out(A)
+	#pragma oss taskiter in(A) out(A) label("iter")
 	for (int i = 0; i < 10; i++) {
-		#pragma oss task in(A)
+		#pragma oss task in(A) label("sleep")
 		sleep_us(10 + A);
-		#pragma oss task out(A)
+		#pragma oss task out(A) label ("add")
 		A = A + 1;
 	}
 
-	# pragma oss task in(A)
+	# pragma oss task in(A) label("print")
 	printf("A=%d\n", A);
+
+	#pragma oss taskwait
+	return 0;
 }