diff --git a/desktop_version/src/Logic.cpp b/desktop_version/src/Logic.cpp
index d460f534..2cdef85b 100644
--- a/desktop_version/src/Logic.cpp
+++ b/desktop_version/src/Logic.cpp
@@ -1664,4 +1664,11 @@ void gamelogic()
     {
         map.glitchname = map.getglitchname(game.roomx, game.roomy);
     }
+
+#if !defined(NO_CUSTOM_LEVELS)
+    if (map.custommode && !map.custommodeforreal && ed.returneditoralpha > 0)
+    {
+        ed.returneditoralpha -= 15;
+    }
+#endif
 }
diff --git a/desktop_version/src/Render.cpp b/desktop_version/src/Render.cpp
index 5dbe5637..82bf1301 100644
--- a/desktop_version/src/Render.cpp
+++ b/desktop_version/src/Render.cpp
@@ -1390,9 +1390,6 @@ void gamerender()
      if(map.custommode && !map.custommodeforreal && !game.advancetext){
         //Return to level editor
         graphics.bprintalpha(5, 5, "[Press ENTER to return to editor]", 220 - (help.glow), 220 - (help.glow), 255 - (help.glow / 2), ed.returneditoralpha, false);
-        if (ed.returneditoralpha > 0) {
-            ed.returneditoralpha -= 15;
-        }
       }
 #endif