diff --git a/cybersyn/scripts/layout.lua b/cybersyn/scripts/layout.lua
index f9eab52..8bbf047 100644
--- a/cybersyn/scripts/layout.lua
+++ b/cybersyn/scripts/layout.lua
@@ -147,7 +147,7 @@ end
 ---@param stop LuaEntity
 ---@param train LuaTrain
 local function get_train_direction(stop, train)
-	local back_rail = train.back_rail
+	local back_rail = train.back_end.rail
 
 	if back_rail then
 		local back_pos = back_rail.position