From 129a5226c5a747d00b88e87f1691a3939eea4f77 Mon Sep 17 00:00:00 2001
From: Chris <blakfyr@gmail.com>
Date: Thu, 23 Oct 2025 21:48:05 -0400
Subject: [PATCH] improv: added reference to movement for UnitComponent.cs

---
 Assets/Scripts/Units/Combat/UnitComponent.cs | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/Assets/Scripts/Units/Combat/UnitComponent.cs b/Assets/Scripts/Units/Combat/UnitComponent.cs
index 6a1d871..66c110e 100644
--- a/Assets/Scripts/Units/Combat/UnitComponent.cs
+++ b/Assets/Scripts/Units/Combat/UnitComponent.cs
@@ -5,18 +5,20 @@ namespace Reset.Units{
         private bool enabledAsHost = true;
 
         private Unit _unit;
-
         internal Unit Unit{
             get {
-                if (_unit != null) {
-                    return _unit;
-                }
-
-                _unit = GetComponent<Unit>();
-
+                if (_unit == null) { _unit = GetComponent<Unit>(); }
                 return _unit;
             }
         }
+        
+        private UnitMovementHandler _movement;
+        internal UnitMovementHandler Movement{
+            get {
+                if (_movement == null) { _movement = GetComponent<UnitMovementHandler>(); }
+                return _movement;
+            }
+        }
 
         void DisableComponent(){
             enabledAsHost = false;