for-the-strength-from-which-he-gives-is-unerring

code/modules/power/gravitygenerator.dm

@@ -179,13 +179,14 @@ var/const/GRAV_NEEDS_WRENCH = 3
 
 // Fixing the gravity generator.
 /obj/machinery/gravity_generator/main/attackby(obj/item/I as obj, mob/user as mob, params)
-	var/old_broken_state = broken_state
 	switch(broken_state)
 		if(GRAV_NEEDS_SCREWDRIVER)
 			if(istype(I, /obj/item/screwdriver))
 				to_chat(user, "<span class='notice'>You secure the screws of the framework.</span>")
 				playsound(src.loc, I.usesound, 50, 1)
 				broken_state++
+				update_icon()
+			return
 		if(GRAV_NEEDS_WELDING)
 			if(istype(I, /obj/item/weldingtool))
 				var/obj/item/weldingtool/WT = I
@@ -193,6 +194,8 @@ var/const/GRAV_NEEDS_WRENCH = 3
 					to_chat(user, "<span class='notice'>You mend the damaged framework.</span>")
 					playsound(src.loc, WT.usesound, 50, 1)
 					broken_state++
+					update_icon()
+			return
 		if(GRAV_NEEDS_PLASTEEL)
 			if(istype(I, /obj/item/stack/sheet/plasteel))
 				var/obj/item/stack/sheet/plasteel/PS = I
@@ -201,17 +204,17 @@ var/const/GRAV_NEEDS_WRENCH = 3
 					to_chat(user, "<span class='notice'>You add the plating to the framework.</span>")
 					playsound(src.loc, PS.usesound, 75, 1)
 					broken_state++
+					update_icon()
 				else
 					to_chat(user, "<span class='notice'>You need 10 sheets of plasteel.</span>")
+			return
 		if(GRAV_NEEDS_WRENCH)
 			if(istype(I, /obj/item/wrench))
 				to_chat(user, "<span class='notice'>You secure the plating to the framework.</span>")
 				playsound(src.loc, I.usesound, 75, 1)
 				set_fix()
-		else
-			..()
-	if(old_broken_state != broken_state)
-		update_icon()
+			return
+	return ..()
 
 /obj/machinery/gravity_generator/main/attack_hand(mob/user as mob)
 	if(!..())