diff --git a/models/Model_Quick96/config_schema.json b/models/Model_Quick96/config_schema.json
index b1d2239..c945a9b 100644
--- a/models/Model_Quick96/config_schema.json
+++ b/models/Model_Quick96/config_schema.json
@@ -12,7 +12,7 @@
         }
       },
       "required": [
-        "batch_size",
+        "batch_size"
       ],
       "title": "dfl_config"
     }
diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py
index 7fb8e00..e1957e7 100644
--- a/samplelib/SampleProcessor.py
+++ b/samplelib/SampleProcessor.py
@@ -266,6 +266,8 @@ class SampleProcessor(object):
                             img_v = np.clip (img_v + (rnd_state.random()-0.5)*a, 0, 1 )
                             img = np.clip( cv2.cvtColor(cv2.merge([img_h, img_s, img_v]), cv2.COLOR_HSV2BGR) , 0, 1 )
                         
+                        img  = imagelib.warp_by_params (warp_params, img,  warp, transform, can_flip=True, border_replicate=border_replicate)
+                        img = np.clip(img.astype(np.float32), 0, 1)
 
                         # Transform from BGR to desired channel_type
                         if channel_type == SPCT.BGR: