Improvements for comments in do...while and debug dump.

1 files changed, 6 insertions, 2 deletions

diff --git a/profile.c b/profile.c
index 4cda2637..e4380280 100644
--- a/profile.c
+++ b/profile.c
@@ -843,9 +843,13 @@ cleanup:
 			indent(SPACEOVER);
 			t1 = pp_pop();
 			fprintf(prof_fp, "} %s (%s)", op2str(Op_K_while), t1->pp_str);
+			if (pc->comment)
+				fprintf(prof_fp, "\t%s", pc->comment->memory->stptr);
+			else {
+				end_line(pc->target_break);
+				skip_comment = true;
+			}
 			pp_free(t1);
-			end_line(pc->target_break);
-			skip_comment = true;
 			pc = pc->target_break;
 			break;