X-Git-Url: https://git.artsoft.org/?a=blobdiff_plain;f=src%2Fscreens.c;h=fbdc116c0787f30b8654f222c190e856bbd91875;hb=6fe98dc77891e9fc45cbd99bbd4793e2133c6960;hp=c0a3b8c64a93b851a9b24380d3cde2c3786cc0b5;hpb=3d97e3d9c20a984e70dae5e63e7c5069fb136c91;p=rocksndiamonds.git

diff --git a/src/screens.c b/src/screens.c
index c0a3b8c6..fbdc116c 100644
--- a/src/screens.c
+++ b/src/screens.c
@@ -58,6 +58,17 @@ void DrawMainMenu()
   int i;
   char *name_text = (!options.network && setup.team_mode ? "Team:" : "Name:");
 
+  /* needed if last screen was the playing screen, invoked from level editor */
+  if (level_editor_test_game)
+  {
+    game_status = LEVELED;
+    DrawLevelEd();
+    return;
+  }
+
+  /* needed if last screen was the level editor */
+  UnmapLevelEditorControlButtons();
+
   FadeSounds();
   GetPlayerConfig();
   LoadLevel(level_nr);