diff --git a/client/src/panels/unitcontrolpanel.ts b/client/src/panels/unitcontrolpanel.ts
index e9a8c3a1..e5d51de0 100644
--- a/client/src/panels/unitcontrolpanel.ts
+++ b/client/src/panels/unitcontrolpanel.ts
@@ -1,4 +1,4 @@
-import { getUnitsManager } from "..";
+import { getMissionData, getUnitsManager } from "..";
 import { Slider } from "../controls/slider";
 import { Unit } from "../units/unit";
 import { Panel } from "./panel";
@@ -70,10 +70,8 @@ export class UnitControlPanel extends Panel {
                 var button = document.createElement("button");
                 button.innerText = unit.getBaseData().unitName;
                 button.setAttribute( "data-short-label", unit.getBaseData().name );
+                button.setAttribute( "data-coalition", unit.getMissionData().coalition );
                 button.classList.add( "pill", "highlight-coalition" )
-                
-                //  TODO: make this dynamic, Mr. Weltro.
-                button.setAttribute( "data-coalition", "blue" );
 
                 button.addEventListener("click", () => getUnitsManager().selectUnit(unit.ID, true));
                 return (button);
diff --git a/client/views/unitcontrolpanel.ejs b/client/views/unitcontrolpanel.ejs
index aed04d5e..37682631 100644
--- a/client/views/unitcontrolpanel.ejs
+++ b/client/views/unitcontrolpanel.ejs
@@ -15,7 +15,7 @@
 
         <div id="selected-units-container" class="ol-scroll">
             <!-- This is where all the unit selection buttons will be shown-->
-            <button class="pill highlight-coalition" data-coalition="blue" data-short-label="18">Olympus 1-1</button>
+            <!-- <button class="pill highlight-coalition" data-coalition="blue" data-short-label="18">Olympus 1-1</button> -->
         </div>
 
         <div id="flight-data">