From 8777b61534e074dd2584dc92f50deba179be5469 Mon Sep 17 00:00:00 2001
From: colaclanth <james.taylor-17@student.manchester.ac.uk>
Date: Tue, 16 Jul 2019 07:40:20 +0100
Subject: [PATCH] Change check when doing final convert to RGB

---
 sstv/decode.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/sstv/decode.py b/sstv/decode.py
index 4ea4f90..270d95e 100644
--- a/sstv/decode.py
+++ b/sstv/decode.py
@@ -136,8 +136,6 @@ class SSTVDecoder(object):
         # vis due to each bit having a length of 30ms. We fix this error margin
         # when decoding the image by aligning each sync pulse
 
-        current_sample = 0
-
         for current_sample in range(0, len(self._samples) - header_size,
                                     jump_size):
             if current_sample % (jump_size * 256) == 0:
@@ -316,7 +314,7 @@ class SSTVDecoder(object):
 
                 pixel_data[x, y] = pixel
 
-        if self.mode.COLOR == spec.COL_FMT.YUV:
+        if image.mode.COLOR != "RGB":
             image = image.convert("RGB")
 
         log_message("...Done!")