From 4931513b8b81428e80b778a7e7956ffbf8bfa75c Mon Sep 17 00:00:00 2001
From: zimzaza4 <3625282098@qq.com>
Date: Thu, 5 Dec 2024 22:36:53 +0800
Subject: [PATCH] fix

---
 .../re/imc/geysermodelenginepackgenerator/GeneratorMain.java  | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/re/imc/geysermodelenginepackgenerator/GeneratorMain.java b/src/main/java/re/imc/geysermodelenginepackgenerator/GeneratorMain.java
index b3437f0..6f8e821 100644
--- a/src/main/java/re/imc/geysermodelenginepackgenerator/GeneratorMain.java
+++ b/src/main/java/re/imc/geysermodelenginepackgenerator/GeneratorMain.java
@@ -344,7 +344,9 @@ public class GeneratorMain {
                     continue;
                 }
                 try {
-                    ImageIO.write(entry.getValue().getImage(), "png", path.toFile());
+                    if (entry.getValue().getImage() != null) {
+                        ImageIO.write(entry.getValue().getImage(), "png", path.toFile());
+                    }
                 } catch (IOException e) {
                     e.printStackTrace();
                 }