From 7f24782aee3d25bef7b9312093b0ea2768cce332 Mon Sep 17 00:00:00 2001
From: DanMB <danielmillst@gmail.com>
Date: Tue, 15 Feb 2022 12:58:08 -0700
Subject: [PATCH] Avoid white screens

---
 .../com/volmit/iris/core/gui/VisionGUI.java    | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/volmit/iris/core/gui/VisionGUI.java b/src/main/java/com/volmit/iris/core/gui/VisionGUI.java
index 42d3dfde3..4677cfc55 100644
--- a/src/main/java/com/volmit/iris/core/gui/VisionGUI.java
+++ b/src/main/java/com/volmit/iris/core/gui/VisionGUI.java
@@ -47,11 +47,7 @@ import javax.imageio.ImageIO;
 import javax.swing.JFrame;
 import javax.swing.JPanel;
 import javax.swing.event.MouseInputListener;
-import java.awt.Color;
-import java.awt.Font;
-import java.awt.Graphics;
-import java.awt.Graphics2D;
-import java.awt.Point;
+import java.awt.*;
 import java.awt.event.KeyEvent;
 import java.awt.event.KeyListener;
 import java.awt.event.MouseEvent;
@@ -436,6 +432,18 @@ public class VisionGUI extends JPanel implements MouseWheelListener, KeyListener
     public void paint(Graphics gx) {
 
         if(engine.isClosed()) {
+            EventQueue.invokeLater(() -> {
+                try
+                {
+                    setVisible(false);
+                }
+
+                catch(Throwable e)
+                {
+
+                }
+            });
+
             return;
         }