1 files changed, 9 insertions, 8 deletions
diff --git a/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.ThreadExtensions/Models/PlotProperties.cs b/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.ThreadExtensions/Models/PlotProperties.cs
index 47632d3aa..ee882ad4b 100644
--- a/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.ThreadExtensions/Models/PlotProperties.cs
+++ b/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.ThreadExtensions/Models/PlotProperties.cs
@@ -5,6 +5,7 @@ using System.Collections.Generic;
 using System.Diagnostics;
 using System.Linq;
 using Tango.BL.Entities;
+using Tango.BL.Enumerations;
 using Tango.Core;
 using Tango.MachineStudio.ThreadExtensions.ViewModels;
 
@@ -15,12 +16,12 @@ namespace Tango.MachineStudio.ThreadExtensions.Models
         public Plot PlotControl { get; set; }
         private IList<DataPoint> _points;
 
-        private string _colorName;
+        private FactorColors _color;
 
-        public string ColorName
+        public FactorColors Color
         {
-            get { return _colorName; }
-            set { _colorName = value; }
+            get { return _color; }
+            set { _color = value; }
         }
 
         /// <summary>
@@ -94,12 +95,12 @@ namespace Tango.MachineStudio.ThreadExtensions.Models
             }
         }
 
-        public PlotProperties(string colorName)
+        public PlotProperties(FactorColors color)
         {
             this.Points = new List<DataPoint>();
             Target100Points = new List<DataPoint>();
             Target200Points = new List<DataPoint>();
-            ColorName = colorName;
+            Color = color;
 
         }
 
@@ -121,8 +122,8 @@ namespace Tango.MachineStudio.ThreadExtensions.Models
             ClearResults();
             PlotControl.InvalidatePlot(true);
 
-            double target100Y = FactorTarget.GetFactor100(ColorName);
-            double target200Y = FactorTarget.GetFactor200(ColorName);
+            double target100Y = FactorTarget.GetFactor100(Color);
+            double target200Y = FactorTarget.GetFactor200(Color);
 
             _to = target100Y > target200Y? target100Y + 10: target200Y + 10;
             _from = target100Y < target200Y ? target100Y - 10 : target200Y - 10;