1 files changed, 5 insertions, 6 deletions
diff --git a/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Developer/ViewModels/MainViewVM.cs b/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Developer/ViewModels/MainViewVM.cs
index cf3cbab7d..e1166a9bc 100644
--- a/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Developer/ViewModels/MainViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Developer/ViewModels/MainViewVM.cs
@@ -28,7 +28,6 @@ using Tango.Transport;
 using Tango.Integration.Printing;
 using Tango.Integration.Diagnostics;
 using Microsoft.Win32;
-using Tango.Integration.Observables.Enumerations;
 
 namespace Tango.MachineStudio.Developer.ViewModels
 {
@@ -210,11 +209,11 @@ namespace Tango.MachineStudio.Developer.ViewModels
             set { _isSideBarOpened = value; RaisePropertyChangedAuto(); }
         }
 
-        private ObservableCollection<Io> _availableGraphs;
+        private ObservableCollection<Graph> _availableGraphs;
         /// <summary>
         /// Gets or sets the available sensors.
         /// </summary>
-        public ObservableCollection<Io> AvailableGraphs
+        public ObservableCollection<Graph> AvailableGraphs
         {
             get { return _availableGraphs; }
             set { _availableGraphs = value; RaisePropertyChangedAuto(); }
@@ -476,7 +475,7 @@ namespace Tango.MachineStudio.Developer.ViewModels
             if (!this.DesignMode)
             {
                 Adapter = ObservablesEntitiesAdapter.Instance;
-                AvailableGraphs = Adapter.Ios.Where(x => (IOType)x.Type == IOType.Input || (IOType)x.Type == IOType.InputOutput).ToObservableCollection();
+                AvailableGraphs = Adapter.Graphs.ToObservableCollection();
             }
 
             Graphs = new ObservableCollection<IRealTimeGraph>();
@@ -1125,7 +1124,7 @@ namespace Tango.MachineStudio.Developer.ViewModels
         /// Add sensor graph from available sensors to displayed graphs.
         /// </summary>
         /// <param name="sensor">The sensor.</param>
-        public void OnDropAvailableGraph(Io sensor)
+        public void OnDropAvailableGraph(Graph sensor)
         {
             if (Graphs.Count < 8)
             {
@@ -1209,7 +1208,7 @@ namespace Tango.MachineStudio.Developer.ViewModels
         /// <param name="graph">The graph.</param>
         public void RemoveGraph(IRealTimeGraph graph)
         {
-            var sensor = graph.Tag as Io;
+            var sensor = graph.Tag as Graph;
             _controllers.Remove(sensor.Name);
             graph.InnerGraph.Dispose();
             Graphs.Remove(graph);