diff --git a/src/org/umlgraph/test/TestUtils.java b/src/org/umlgraph/test/TestUtils.java
index 668b5cd..ec8a02c 100644
--- a/src/org/umlgraph/test/TestUtils.java
+++ b/src/org/umlgraph/test/TestUtils.java
@@ -68,6 +68,9 @@ public class TestUtils {
 		} else if (refLine.startsWith("<!-- Generated by javadoc ")) {
 		    if (!outFile.startsWith("<!-- Generated by javadoc "))
 			break;
+		} else if (refLine.startsWith("<!-- UML diagram added by UMLGraph")) {
+		    if (!outFile.startsWith("<!-- UML diagram added by UMLGraph"))
+			break;
 		} else if (!refLine.equals(outFile)) {
 		    break;
 		}