1 files changed, 16 insertions, 1 deletions
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/Controls/MachineConnectionIcon.xaml.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/Controls/MachineConnectionIcon.xaml.cs
index 72b9af55f..6f754e306 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/Controls/MachineConnectionIcon.xaml.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Common/Controls/MachineConnectionIcon.xaml.cs
@@ -32,6 +32,14 @@ namespace Tango.MachineStudio.Common.Controls
     /// </summary>
     public partial class MachineConnectionIcon : UserControl
     {
+        public MachineTypes MachineType
+        {
+            get { return (MachineTypes)GetValue(MachineTypeProperty); }
+            set { SetValue(MachineTypeProperty, value); }
+        }
+        public static readonly DependencyProperty MachineTypeProperty =
+            DependencyProperty.Register("MachineType", typeof(MachineTypes), typeof(MachineConnectionIcon), new PropertyMetadata(MachineTypes.TS1800));
+
         public IExternalBridgeClient ExternalBridgeClient
         {
             get { return (IExternalBridgeClient)GetValue(ExternalBridgeClientProperty); }
@@ -56,13 +64,20 @@ namespace Tango.MachineStudio.Common.Controls
                 {
                     ExternalBridgeClientType = ExternalBridgeClientType.Usb;
                 }
-                else if (ExternalBridgeClient.GetType() == typeof(ExternalBridgeTcpClient))
+                else if (ExternalBridgeClient.GetType() == typeof(ExternalBridgeTcpFirmwareClient))
                 {
                     ExternalBridgeClientType = ExternalBridgeClientType.Tcp;
+                    MachineType = MachineTypes.Eureka;
+                }
+                else if (ExternalBridgeClient.GetType() == typeof(ExternalBridgeTcpClient))
+                {
+                    ExternalBridgeClientType = ExternalBridgeClientType.Wifi;
+                    MachineType = ExternalBridgeClient.MachineType;
                 }
                 else if (ExternalBridgeClient.GetType() == typeof(ExternalBridgeSignalRClient))
                 {
                     ExternalBridgeClientType = ExternalBridgeClientType.SignalR;
+                    MachineType = ExternalBridgeClient.MachineType;
                 }
                 else if (ExternalBridgeClient.GetType() == typeof(EmulatorExternalBridge))
                 {