diff --git a/core/imagelib/warp.py b/core/imagelib/warp.py
index 4f6f68a..4de56d2 100644
--- a/core/imagelib/warp.py
+++ b/core/imagelib/warp.py
@@ -104,7 +104,7 @@ def gen_pts(W, H, rnd_state=None):
     return pts1, pts2
     
     
-def gen_warp_params (w, flip=False, rotation_range=[-2,2], scale_range=[-0.5, 0.5], tx_range=[-0.05, 0.05], ty_range=[-0.05, 0.05], rnd_state=None  ):
+def gen_warp_params (w, flip=False, rotation_range=[-2,2], scale_range=[-0.5, 0.5], tx_range=[-0.05, 0.05], ty_range=[-0.05, 0.05], rnd_state=None, warp_rnd_state=None):
     if rnd_state is None:
         rnd_state = np.random
     if warp_rnd_state is None:
@@ -178,4 +178,4 @@ def warp_by_params (params, img, can_warp, can_transform, can_flip, border_repli
         img = img[...,None]
     if can_flip and params['flip']:
         img = img[:,::-1,...]
-    return img
\ No newline at end of file
+    return img