diff --git a/NEWS b/NEWS
index 3d5be83aa..e85f4fc60 100644
--- a/NEWS
+++ b/NEWS
@@ -8,6 +8,7 @@ DFHack Future
     New tweaks
     New features
     Fixes
+        steam-engine: fixed a crash on arena load
     Misc Improvements
     Removed
 
diff --git a/plugins/steam-engine.cpp b/plugins/steam-engine.cpp
index 2a22d5953..cc9455914 100644
--- a/plugins/steam-engine.cpp
+++ b/plugins/steam-engine.cpp
@@ -5,6 +5,7 @@
 #include <modules/Gui.h>
 #include <modules/Screen.h>
 #include <modules/Maps.h>
+#include <modules/World.h>
 #include <TileTypes.h>
 #include <vector>
 #include <cstdio>
@@ -978,8 +979,8 @@ static void enable_hooks(bool enable)
 DFhackCExport command_result plugin_onstatechange(color_ostream &out, state_change_event event)
 {
     switch (event) {
-    case SC_WORLD_LOADED:
-        if (find_engines(out))
+    case SC_MAP_LOADED:
+        if (World::isFortressMode() && find_engines(out))
         {
             out.print("Detected steam engine workshops - enabling plugin.\n");
             enable_hooks(true);
@@ -987,7 +988,7 @@ DFhackCExport command_result plugin_onstatechange(color_ostream &out, state_chan
         else
             enable_hooks(false);
         break;
-    case SC_WORLD_UNLOADED:
+    case SC_MAP_UNLOADED:
         enable_hooks(false);
         engines.clear();
         break;