1 files changed, 20 insertions, 0 deletions
diff --git a/go/game/unit.go b/go/game/unit.go
index 94458c07..adb608ab 100644
--- a/go/game/unit.go
+++ b/go/game/unit.go
@@ -251,3 +251,23 @@ func (u *Unit) onUnpile(containing Permanent) {
 func (u *Unit) onDrop(containing Permanent) {
 	u.onUnpile(containing)
 }
+
+func (u *Unit) removeFullAction(tag string) {
+	for i, fa := range u.FullActions {
+		if fa.tag != tag {
+			continue
+		}
+		u.FullActions[i] = u.FullActions[len(u.FullActions)-1]
+		u.FullActions = u.FullActions[:len(u.FullActions)-1]
+		break
+	}
+}
+
+func (u *Unit) AddDamage(damage int) {
+	if u.Marks(UnitMarks.Ward) > 0 {
+		u.adjustMarks(UnitMarks.Ward, -1)
+		return
+	}
+
+	u.PermanentBase.AddDamage(damage)
+}