diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py
index eb515f9..2cfc8b3 100644
--- a/samplelib/SampleProcessor.py
+++ b/samplelib/SampleProcessor.py
@@ -200,7 +200,7 @@ class SampleProcessor(object):
                                 
                         # Apply random color transfer                        
                         if ct_mode is not None and ct_sample is not None or ct_mode == 'fs-aug':
-                            if 'fs-aug':
+                            if ct_mode == 'fs-aug':
                                 img = imagelib.color_augmentation(img)
                             else:
                                 if ct_sample_bgr is None: