diff --git a/src/mudlib/server.py b/src/mudlib/server.py
index 418163a..3d93dee 100644
--- a/src/mudlib/server.py
+++ b/src/mudlib/server.py
@@ -45,6 +45,7 @@ from mudlib.if_session import broadcast_to_spectators
 from mudlib.mob_ai import process_mobs
 from mudlib.mobs import load_mob_templates, mob_templates
 from mudlib.player import Player, players
+from mudlib.prompt import render_prompt
 from mudlib.resting import process_resting
 from mudlib.store import (
     PlayerData,
@@ -391,10 +392,8 @@ async def shell(
             # Show appropriate prompt based on mode
             if player.mode == "editor" and player.editor:
                 _writer.write(f"  {player.editor.cursor + 1}> ")
-            elif player.mode == "if" and player.if_session:
-                _writer.write("> ")
             else:
-                _writer.write("mud> ")
+                _writer.write(render_prompt(player))
             await _writer.drain()
 
             inp = await readline2(_reader, _writer)