From 46d21ae2c218a2f86897b4d71970787cb5a5bfd4 Mon Sep 17 00:00:00 2001
From: paprka <ddkfjskdjf@gayshit.com>
Date: Fri, 13 Feb 2015 17:48:58 -0800
Subject: [PATCH] robots lose power from their cell for leaving their baton
 active

---
 code/game/objects/items/weapons/stunbaton.dm | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/code/game/objects/items/weapons/stunbaton.dm b/code/game/objects/items/weapons/stunbaton.dm
index 8ae2906fa5f..81ad1f2f4c9 100644
--- a/code/game/objects/items/weapons/stunbaton.dm
+++ b/code/game/objects/items/weapons/stunbaton.dm
@@ -57,7 +57,12 @@
 			status = 0
 			update_icon()
 		if(status)
-			bcell.use(losspertick)
+			if(isrobot(loc))
+				var/mob/living/silicon/robot/R = loc
+				if(R && R.cell)
+					R.cell.use(losspertick)
+			else
+				bcell.use(losspertick)
 			update_icon()
 
 /obj/item/weapon/melee/baton/update_icon()