diff --git a/src/Umbraco.Tests/Web/TemplateUtilitiesTests.cs b/src/Umbraco.Tests/Web/TemplateUtilitiesTests.cs
index 3a5405548b..c0a6e97dda 100644
--- a/src/Umbraco.Tests/Web/TemplateUtilitiesTests.cs
+++ b/src/Umbraco.Tests/Web/TemplateUtilitiesTests.cs
@@ -99,7 +99,7 @@ namespace Umbraco.Tests.Web
             var snapshotService = Mock.Of<IPublishedSnapshotService>();
             Mock.Get(snapshotService).Setup(x => x.CreatePublishedSnapshot(It.IsAny<string>())).Returns(snapshot);
             var media = Mock.Of<IPublishedContent>();
-            Mock.Get(media).Setup(x => x.Url).Returns("/media/1001/my-image.jpg");
+            Mock.Get(media).Setup(x => x.Url(It.IsAny<string>(), It.IsAny<UrlMode>())).Returns("/media/1001/my-image.jpg");
             var mediaCache = Mock.Of<IPublishedMediaCache>();
             Mock.Get(mediaCache).Setup(x => x.GetById(It.IsAny<Guid>())).Returns(media);
 
diff --git a/src/Umbraco.Web/Templates/TemplateUtilities.cs b/src/Umbraco.Web/Templates/TemplateUtilities.cs
index 040c4e128a..06706e475b 100644
--- a/src/Umbraco.Web/Templates/TemplateUtilities.cs
+++ b/src/Umbraco.Web/Templates/TemplateUtilities.cs
@@ -60,7 +60,7 @@ namespace Umbraco.Web.Templates
                             if (guidUdi.EntityType == Constants.UdiEntityType.Document)
                                 newLink = urlProvider.GetUrl(guidUdi.Guid);
                             else if (guidUdi.EntityType == Constants.UdiEntityType.Media)
-                                newLink = mediaCache.GetById(guidUdi.Guid)?.Url;
+                                newLink = mediaCache.GetById(guidUdi.Guid)?.Url();
 
                             if (newLink == null)
                                 newLink = "#";