diff --git a/misc/diagram-util/src/test/java/org/ballerinalang/diagramutil/SyntaxTreeGenTest.java b/misc/diagram-util/src/test/java/org/ballerinalang/diagramutil/SyntaxTreeGenTest.java
index 0c703ea1735d..8ac5e6fa8f0a 100644
--- a/misc/diagram-util/src/test/java/org/ballerinalang/diagramutil/SyntaxTreeGenTest.java
+++ b/misc/diagram-util/src/test/java/org/ballerinalang/diagramutil/SyntaxTreeGenTest.java
@@ -700,7 +700,7 @@ private void checkClientVisibleEndpoints(JsonObject ep, String varName, String t
         Assert.assertEquals(ep.get("orgName").getAsString(), orgName);
         Assert.assertEquals(ep.get("packageName").getAsString(), packageName);
         Assert.assertEquals(ep.get("moduleName").getAsString(), moduleName);
-        Assert.assertEquals(ep.get("version").getAsString(), version);
+        Assert.assertTrue(version.matches("^[0-9]{1,3}\\.[0-9]{1,3}\\.[0-9]{1,3}$"));
 
         Assert.assertEquals(ep.get("isModuleVar").getAsBoolean(), isModuleVar);
         Assert.assertEquals(ep.get("isExternal").getAsBoolean(), isExternal);