diff --git a/cicaprojekt/Game.java b/cicaprojekt/Game.java
index c47019a..ff9b040 100644
--- a/cicaprojekt/Game.java
+++ b/cicaprojekt/Game.java
@@ -1,5 +1,6 @@
 package cicaprojekt;
 
+import javax.swing.*;
 import java.awt.*;
 import java.io.File;
 import java.io.IOException;
@@ -105,6 +106,12 @@ public class Game {
     }
 
     public void updateDisplay() {
-        display.repaint();
+        try {
+            display.drawVisuals();
+            display.repaint();
+        }
+        catch (IOException e) {
+            Control.ioErrorMessage();
+        }
     }
 }