diff --git a/src/program.py b/src/program.py
index 34a8e42..30bb04c 100644
--- a/src/program.py
+++ b/src/program.py
@@ -41,7 +41,6 @@ layout = [
 window = sg.Window("Model Folder", layout)
 while True:
     event, values = window.read()
-    print(event)
     if event == sg.WIN_CLOSED or event == "Cancel":
         break
     if event == "model_folder":
@@ -58,11 +57,17 @@ while True:
     if event == "models":
         model_path = values["models"]
         preview_image, civitai_info = get_extra_filenames(model_path)
-        img = Image.open(preview_image)
-        img.thumbnail((512, 512))
-        bio = io.BytesIO()
-        img.save(bio, format="PNG")
-        window["preview_image"].update(data=bio.getvalue())
-        window["civitai_info"].update(get_civitai_text(civitai_info))
+        if preview_image:
+            img = Image.open(preview_image)
+            img.thumbnail((512, 512))
+            bio = io.BytesIO()
+            img.save(bio, format="PNG")
+            window["preview_image"].update(data=bio.getvalue())
+        else:
+            window["preview_image"].update(data=None)
+        if civitai_info:
+            window["civitai_info"].update(get_civitai_text(civitai_info))
+        else:
+            window["civitai_info"].update(data=None)
 
 window.close()