From 7fd050e6d8190e0eb07d03135ee03a99a1e04637 Mon Sep 17 00:00:00 2001
From: kingarthur91 <arthurkoy_durbin@yahoo.com>
Date: Sun, 26 Nov 2023 03:54:13 -0500
Subject: [PATCH] fix train factory crash

---
 changelog.txt                          | 1 +
 prototypes/functions/compatibility.lua | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/changelog.txt b/changelog.txt
index c5c1a5e..be223fa 100644
--- a/changelog.txt
+++ b/changelog.txt
@@ -3,6 +3,7 @@ Version: 0.2.16
 Date: ?
   Changes:
     - fix crash with transport drones compatibility
+    - fix crash with train factory mod
 ---------------------------------------------------------------------------------------------------
 Version: 0.2.15
 Date: 2023-11-21
diff --git a/prototypes/functions/compatibility.lua b/prototypes/functions/compatibility.lua
index d62a1e8..61e0c3a 100644
--- a/prototypes/functions/compatibility.lua
+++ b/prototypes/functions/compatibility.lua
@@ -227,6 +227,10 @@ if mods['Transport_Drones'] then
     end
 end
 
+if mods['trainfactory'] then
+    data.raw['train-stop']['trainfactory-trainstop'].next_upgrade = nil
+end
+
 if mods['miniloader'] then
     TECHNOLOGY('miniloader'):add_pack('py-science-pack-1'):add_pack('logistic-science-pack')
     TECHNOLOGY('fast-miniloader'):add_pack('py-science-pack-2')