Fixes wormhole projectors (#36622)

2018-03-26 17:19:46 -05:00
parent c579d81f4c
commit f8697faf8f
3 changed files with 17 additions and 2 deletions
@@ -185,7 +185,16 @@
 /obj/item/gun/energy/wormhole_projector/update_icon()
 	icon_state = "[initial(icon_state)][select]"
 	item_state = icon_state
-	return
+
+/obj/item/gun/energy/wormhole_projector/update_ammo_types()
+	. = ..()
+	for(var/i in 1 to ammo_type.len)
+		var/obj/item/ammo_casing/energy/wormhole/W = ammo_type[i]
+		if(istype(W))
+			W.gun = src
+			var/obj/item/projectile/beam/wormhole/WH = W.BB
+			if(istype(WH))
+				WH.gun = src

 /obj/item/gun/energy/wormhole_projector/process_chamber()
 	..()