From ac675d14e952788881e4980e0f20c57aae967d9c Mon Sep 17 00:00:00 2001
From: Sebastiaan Janssen <sebastiaan@umbraco.com>
Date: Wed, 22 Jan 2020 11:13:49 +0100
Subject: [PATCH] Fixes build errors

---
 src/Umbraco.Core/ContentVariationExtensions.cs | 13 ++++---------
 src/Umbraco.TestData/SegmentTestController.cs  |  8 ++++----
 2 files changed, 8 insertions(+), 13 deletions(-)

diff --git a/src/Umbraco.Core/ContentVariationExtensions.cs b/src/Umbraco.Core/ContentVariationExtensions.cs
index 8939f7133b..4fc6f418aa 100644
--- a/src/Umbraco.Core/ContentVariationExtensions.cs
+++ b/src/Umbraco.Core/ContentVariationExtensions.cs
@@ -30,6 +30,10 @@ namespace Umbraco.Core
         /// <summary>
         /// Determines whether the content type varies by segment.
         /// </summary>
+        /// <param name="contentType">The content type.</param>
+        /// <returns>
+        /// A value indicating whether the content type varies by segment.
+        /// </returns>
         public static bool VariesBySegment(this ISimpleContentType contentType) => contentType.Variations.VariesBySegment();
 
         /// <summary>
@@ -122,15 +126,6 @@ namespace Umbraco.Core
         /// </returns>
         public static bool VariesByCulture(this ContentVariation variation) => (variation & ContentVariation.Culture) > 0;
 
-        /// <summary>
-        /// Determines whether the content type varies by segment.
-        /// </summary>
-        /// <param name="contentType">The content type.</param>
-        /// <returns>
-        /// A value indicating whether the content type varies by segment.
-        /// </returns>
-        public static bool VariesBySegment(this ISimpleContentType contentType) => contentType.Variations.VariesBySegment();
-
         /// <summary>
         /// Determines whether the content type varies by segment.
         /// </summary>
diff --git a/src/Umbraco.TestData/SegmentTestController.cs b/src/Umbraco.TestData/SegmentTestController.cs
index 33badbbb55..c78d12ab0e 100644
--- a/src/Umbraco.TestData/SegmentTestController.cs
+++ b/src/Umbraco.TestData/SegmentTestController.cs
@@ -30,9 +30,9 @@ namespace Umbraco.TestData
             if (ct.Variations.VariesBySegment())
                 return Content($"The document type {alias} already allows segments, nothing has been changed");
 
-            ct.Variations = ct.Variations.SetFlag(ContentVariation.Segment);
-            
-            propType.Variations = propType.Variations.SetFlag(ContentVariation.Segment);
+            ct.Variations = ct.Variations.SetVariesBy(ContentVariation.Segment);
+
+            propType.Variations = propType.Variations.SetVariesBy(ContentVariation.Segment);
 
             Services.ContentTypeService.Save(ct);
             return Content($"The document type {alias} and property type {propertyTypeAlias} now allows segments");
@@ -50,7 +50,7 @@ namespace Umbraco.TestData
             if (!ct.VariesBySegment())
                 return Content($"The document type {alias} does not allow segments, nothing has been changed");
 
-            ct.Variations = ct.Variations.UnsetFlag(ContentVariation.Segment);
+            ct.Variations = ct.Variations.SetVariesBy(ContentVariation.Nothing);
 
             Services.ContentTypeService.Save(ct);
             return Content($"The document type {alias} no longer allows segments");