From 06fff9a83f85cc01b01921aa92ddf20b88c40a59 Mon Sep 17 00:00:00 2001
From: Aokromes <aokromes@gmail.com>
Date: Fri, 21 Oct 2016 07:26:23 +0200
Subject: [PATCH] Revert "Core/Unit: Remove unused code as for 4.3.4"

This reverts commit 42e74822b8760e5842299185c2cda7d08b2d9e3e.
---
 src/server/game/Entities/Unit/Unit.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/server/game/Entities/Unit/Unit.cpp b/src/server/game/Entities/Unit/Unit.cpp
index b3bf984b37..f3752147f3 100644
--- a/src/server/game/Entities/Unit/Unit.cpp
+++ b/src/server/game/Entities/Unit/Unit.cpp
@@ -5329,6 +5329,10 @@ void Unit::setPowerType(Powers new_powertype)
         case POWER_ENERGY:
             SetMaxPower(POWER_ENERGY, uint32(std::ceil(GetCreatePowers(POWER_ENERGY) * powerMultiplier)));
             break;
+        case POWER_HAPPINESS:
+            SetMaxPower(POWER_HAPPINESS, uint32(std::ceil(GetCreatePowers(POWER_HAPPINESS) * powerMultiplier)));
+            SetPower(POWER_HAPPINESS, uint32(std::ceil(GetCreatePowers(POWER_HAPPINESS) * powerMultiplier)));
+            break;
     }
 }