From 94279fabb2615bc955dfa05eaab9a92c705a4ff1 Mon Sep 17 00:00:00 2001
From: Jesper Christensen Mayntzhusen <jma@limbo.works>
Date: Wed, 28 Apr 2021 20:57:17 +0200
Subject: [PATCH] support IPublishedElement

---
 src/Umbraco.ModelsBuilder/Building/CodeParser.cs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/Umbraco.ModelsBuilder/Building/CodeParser.cs b/src/Umbraco.ModelsBuilder/Building/CodeParser.cs
index 30fcbf1f..ababfce9 100644
--- a/src/Umbraco.ModelsBuilder/Building/CodeParser.cs
+++ b/src/Umbraco.ModelsBuilder/Building/CodeParser.cs
@@ -78,7 +78,10 @@ private static void Parse(ParseResult disco, CSharpCompilation compilation, Synt
                         if (x.Parameters.Length != 1) return false;
                         var type1 = x.Parameters[0].Type;
                         var type2 = typeof (IPublishedContent);
-                        return type1.ToDisplayString() == type2.FullName;
+                        var type3 = typeof(IPublishedElement);
+                        if (type1.ToDisplayString() == type2.FullName || type1.ToDisplayString() == type3.FullName) return true;
+
+                        return false;
                     });
 
                 if (hasCtor)