diff --git a/survival/game/user_interface.py b/survival/game/user_interface.py
index b1acb76..9887f9f 100644
--- a/survival/game/user_interface.py
+++ b/survival/game/user_interface.py
@@ -25,9 +25,11 @@ class UserInterface:
             i += 1
         self.slot_image = Image('ui.png', self.pos, scale=2)
         self.font = pygame.font.SysFont('Comic Sans MS', 20)
+        self.initialized = False
 
     def load_inventory(self, inventory: InventoryComponent):
         self.inventory = inventory
+        self.initialized = True
 
     def update(self):
         pass
diff --git a/survival/generators/world_generator.py b/survival/generators/world_generator.py
index 09e58ee..975944c 100644
--- a/survival/generators/world_generator.py
+++ b/survival/generators/world_generator.py
@@ -51,8 +51,8 @@ class WorldGenerator:
         self.world.add_processor(VisionSystem(self.camera))
 
         self.player = PlayerGenerator().create_player(self.world, self.game_map)
-        self.world.get_processor(DrawSystem).initialize_interface(
-            self.world.component_for_entity(self.player, InventoryComponent))
+        # self.world.get_processor(DrawSystem).initialize_interface(
+        #     self.world.component_for_entity(self.player, InventoryComponent))
 
         # BuildingGenerator().create_home(self.world, self.game_map)
         self.resource_generator.generate_resources(self.player)
diff --git a/survival/systems/draw_system.py b/survival/systems/draw_system.py
index 37de885..223d8fb 100644
--- a/survival/systems/draw_system.py
+++ b/survival/systems/draw_system.py
@@ -17,5 +17,6 @@ class DrawSystem(esper.Processor):
             sprite.image.pos = pos.position
             sprite.image.origin = (32 * pos.direction.value, 0)
             self.camera.draw(sprite.image)
-            self.ui.update()
-            self.ui.draw()
+            if self.ui.initialized:
+                self.ui.update()
+                self.ui.draw()