diff --git a/app/src/main/kotlin/com/simplemobiletools/draw/activities/MainActivity.kt b/app/src/main/kotlin/com/simplemobiletools/draw/activities/MainActivity.kt
index 7539ec5..d6339a8 100644
--- a/app/src/main/kotlin/com/simplemobiletools/draw/activities/MainActivity.kt
+++ b/app/src/main/kotlin/com/simplemobiletools/draw/activities/MainActivity.kt
@@ -285,8 +285,9 @@ class MainActivity : SimpleActivity(), CanvasListener {
     }
 
     private fun saveImage() {
-        SaveImageDialog(this, suggestedFileExtension, curPath, my_canvas) {
-            curPath = it
+        SaveImageDialog(this, suggestedFileExtension, curPath, my_canvas) { path, extension ->
+            curPath = path
+            suggestedFileExtension = extension
         }
     }
 
diff --git a/app/src/main/kotlin/com/simplemobiletools/draw/dialogs/SaveImageDialog.kt b/app/src/main/kotlin/com/simplemobiletools/draw/dialogs/SaveImageDialog.kt
index fa01199..a0c11a9 100644
--- a/app/src/main/kotlin/com/simplemobiletools/draw/dialogs/SaveImageDialog.kt
+++ b/app/src/main/kotlin/com/simplemobiletools/draw/dialogs/SaveImageDialog.kt
@@ -15,7 +15,8 @@ import kotlinx.android.synthetic.main.dialog_save_image.view.*
 import java.io.File
 import java.io.OutputStream
 
-class SaveImageDialog(val activity: SimpleActivity, val suggestedExtension: String, val curPath: String, val canvas: MyCanvas, callback: (path: String) -> Unit) {
+class SaveImageDialog(val activity: SimpleActivity, val suggestedExtension: String, val curPath: String, val canvas: MyCanvas,
+                      callback: (path: String, extension: String) -> Unit) {
     private val SIMPLE_DRAW = "Simple Draw"
 
     init {
@@ -64,7 +65,7 @@ class SaveImageDialog(val activity: SimpleActivity, val suggestedExtension: Stri
                     }
 
                     if (saveFile(newFile)) {
-                        callback(newFile.absolutePath)
+                        callback(newFile.absolutePath, extension)
                         dismiss()
                     } else {
                         activity.toast(R.string.unknown_error_occurred)