From afba1628c2ff20c8c84dafaa468b7ee97f2ee3b8 Mon Sep 17 00:00:00 2001
From: amastis <37512516+amastis@users.noreply.github.com>
Date: Mon, 31 May 2021 12:34:28 -0400
Subject: [PATCH] fixed extension handling if not lossy_formats

---
 LSBSteg.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/LSBSteg.py b/LSBSteg.py
index 895c65a..90d9902 100755
--- a/LSBSteg.py
+++ b/LSBSteg.py
@@ -181,6 +181,8 @@ def main():
         if out_ext in lossy_formats:
             out_f = out_f + ".png"
             print("Output file changed to ", out_f)
+        else:
+            out_f = out_f + "." + out_ext
 
         data = open(args["--file"], "rb").read()
         res = steg.encode_binary(data)