diff --git a/dungeonbattle/display/display.py b/dungeonbattle/display/display.py
index de5c280..b0ced92 100644
--- a/dungeonbattle/display/display.py
+++ b/dungeonbattle/display/display.py
@@ -7,11 +7,12 @@ class Display:
     def __init__(self, game, screen):
         self.screen = screen
         self.game = game
+        lines = curses.LINES if screen else 4
+        cols = curses.COLS * 4 // 5 if screen else 4
         self.map_display = MapDisplay(game.m,
                                       TexturePack.get_pack(
                                          game.settings.TEXTURE_PACK),
-                                      curses.LINES,
-                                      curses.COLS * 4 // 5)
+                                      lines, cols, screen is not None)
 
     def refresh(self):
         self.map_display.update_pad()
diff --git a/dungeonbattle/tests/game_test.py b/dungeonbattle/tests/game_test.py
index 4183cbe..f4c1fdf 100644
--- a/dungeonbattle/tests/game_test.py
+++ b/dungeonbattle/tests/game_test.py
@@ -10,7 +10,7 @@ class TestGame(unittest.TestCase):
         Setup game.
         """
         self.game = Game()
-        self.game.new_game(False)
+        self.game.new_game(None)
 
     def test_load_game(self) -> None:
         self.assertRaises(NotImplementedError, Game.load_game, "game.save")