Fix bug where mode optional argument did not function properly

Aqcurate · May 14, 2018 · 92a2aa4 · 92a2aa4
1 parent 3e1c7a7
commit 92a2aa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/steglsb.py b/steglsb.py
@@ -49,7 +49,7 @@ def __init__(self, cover, secret, bits, outfile, mode=None):
         self.outfile = outfile
         self.cover = self._get_image(cover, 'cover')
         if mode != None:
-            self.cover.convert(mode)
+            self.cover = self.cover.convert(mode.upper())
             print('[*] Converted cover image mode to {}.'.format(self.cover.mode))
         self.secret = self._get_image(secret, 'secret').convert(self.cover.mode)
         print('[*] Converted secret image mode to {}.'.format(self.cover.mode))