diff --git a/save_screen.sh b/save_screen.sh
index 1a1fe7e..e670a2e 100755
--- a/save_screen.sh
+++ b/save_screen.sh
@@ -1,6 +1,6 @@
 #!/bin/sh
-filename=$(basename "$1")
-file=$(zenity --file-selection --save --filename="$filename")
+filename=$(basename -s ".html" "$1")
+file=$(zenity --file-selection --save --filename="$filename.html")
 if [ -n "$file" ]; then
     cat > "$file"
 fi