diff --git a/code/game/machinery/computer/law.dm b/code/game/machinery/computer/law.dm
index 125a5fc06ad..6541c897447 100644
--- a/code/game/machinery/computer/law.dm
+++ b/code/game/machinery/computer/law.dm
@@ -29,10 +29,11 @@
 
 
 	attackby(obj/item/weapon/O as obj, mob/user as mob, params)
-		if(is_away_level(user.z))
-			to_chat(user, "<span class='danger'>Unable to establish a connection</span>: You're too far away from the station!")
-			return
 		if(istype(O, /obj/item/weapon/aiModule))
+			var/turf/T = get_turf(current)
+			if(!atoms_share_level(T, src))
+				to_chat(user, "<span class='danger'>Unable to establish a connection</span>: You're too far away from the target silicon!")
+				return
 			var/datum/game_mode/nations/mode = get_nations_mode()
 			if(!mode)
 				var/obj/item/weapon/aiModule/M = O
@@ -74,6 +75,10 @@
 
 	attackby(obj/item/weapon/aiModule/module as obj, mob/user as mob, params)
 		if(istype(module, /obj/item/weapon/aiModule))
+			var/turf/T = get_turf(current)
+			if(!atoms_share_level(T, src))
+				to_chat(user, "<span class='danger'>Unable to establish a connection</span>: You're too far away from the target silicon!")
+				return
 			var/datum/game_mode/nations/mode = get_nations_mode()
 			if(!mode)
 				module.install(src)
diff --git a/code/modules/computer3/computers/law.dm b/code/modules/computer3/computers/law.dm
index 533a600a4eb..e36aca935ec 100644
--- a/code/modules/computer3/computers/law.dm
+++ b/code/modules/computer3/computers/law.dm
@@ -24,10 +24,11 @@
 
 
 	attackby(obj/item/weapon/aiModule/module as obj, mob/user as mob, params)
-		if(is_away_level(user.z))
-			to_chat(user, "<span class='danger'>Unable to establish a connection</span>: You're too far away from the station!")
-			return
 		if(istype(module, /obj/item/weapon/aiModule))
+			var/turf/T = get_turf(current)
+			if(!atoms_share_level(T, src))
+				to_chat(user, "<span class='danger'>Unable to establish a connection</span>: You're too far away from the target silicon!")
+				return
 			module.install(src)
 		else
 			return ..()
@@ -60,6 +61,10 @@
 
 	attackby(obj/item/weapon/aiModule/module as obj, mob/user as mob, params)
 		if(istype(module, /obj/item/weapon/aiModule))
+			var/turf/T = get_turf(current)
+			if(!atoms_share_level(T, src))
+				to_chat(user, "<span class='danger'>Unable to establish a connection</span>: You're too far away from the target silicon!")
+				return
 			module.install(src)
 		else
 			return ..()