From c80b810118c873d390ca7a7b4eac9230d6f02dde Mon Sep 17 00:00:00 2001
From: Rockdtben <ben@benjamingare.com>
Date: Mon, 29 Jan 2018 19:04:03 -0600
Subject: [PATCH] Fixes issue #15

---
 src/main/java/com/github/monster860/fastdmm/FastDMM.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/main/java/com/github/monster860/fastdmm/FastDMM.java b/src/main/java/com/github/monster860/fastdmm/FastDMM.java
index 7ff4b5e..49a371b 100644
--- a/src/main/java/com/github/monster860/fastdmm/FastDMM.java
+++ b/src/main/java/com/github/monster860/fastdmm/FastDMM.java
@@ -502,6 +502,7 @@ public void actionPerformed(ActionEvent e) {
 			}
 		} else if("mapimage".equals(e.getActionCommand())) {
 			JFileChooser fc = new JFileChooser();
+			
 			if (fc.getChoosableFileFilters().length > 0)
 				fc.removeChoosableFileFilter(fc.getChoosableFileFilters()[0]);
 			fc.addChoosableFileFilter(new FileNameExtensionFilter("Image (*.png)", "png"));
@@ -512,6 +513,9 @@ public void actionPerformed(ActionEvent e) {
 					mapImage = createMapImage(1);
 				}
 				try {
+					if (!fc.getSelectedFile().getPath().endsWith(".png"))
+						fc.setSelectedFile(new File(fc.getSelectedFile().getPath() + ".png"));					
+					
 					ImageIO.write(mapImage, "png", fc.getSelectedFile());
 				} catch (IOException e1) {
 					e1.printStackTrace();