diff --git a/trains/binding/frameworks/lightgbm_bind.py b/trains/binding/frameworks/lightgbm_bind.py
index e8bf116c..22d0ec1b 100644
--- a/trains/binding/frameworks/lightgbm_bind.py
+++ b/trains/binding/frameworks/lightgbm_bind.py
@@ -123,9 +123,10 @@ class PatchLIGHTgbmModelIO(PatchBaseModelIO):
         ret = original_fn(*args, **kwargs)
         if not PatchLIGHTgbmModelIO.__main_task:
             return ret
-        params = args[0] if args else kwargs['params']
+        params = args[0] if args else kwargs.get('params', {})
         for k, v in params.items():
             if isinstance(v, set):
                 params[k] = list(v)
-        PatchLIGHTgbmModelIO.__main_task.connect(params)
+        if params:
+            PatchLIGHTgbmModelIO.__main_task.connect(params)
         return ret