diff --git a/GoldWars/GoldWars/Entities/Base.swift b/GoldWars/GoldWars/Entities/Base.swift
index 6851cf3..e8384bd 100644
--- a/GoldWars/GoldWars/Entities/Base.swift
+++ b/GoldWars/GoldWars/Entities/Base.swift
@@ -32,11 +32,12 @@ class Base: GKEntity {
     }
     
 	func attackBase(base: Base, units:Int) -> [GKEntity]{
-			base.changeOwnerShip = true
-			self.unitCount -= units
-			base.unitCount += units
-			return [self, base]
-	}
+        base.changeOwnerShip = true
+        base.ownerShipPlayer = self.ownerShipPlayer
+        self.unitCount -= units
+        base.unitCount += units
+        return [self, base]
+    }
     
     required init?(coder: NSCoder) {
         fatalError("init(coder:) has not been implemented")