diff --git a/setup.py b/setup.py
index e57e2bb..7da2228 100644
--- a/setup.py
+++ b/setup.py
@@ -15,11 +15,11 @@
 
 setup(name='steppy-toolkit',
       packages=find_packages(),
-      version='0.1.11',
+      version='0.1.12',
       description='Set of tools to make your work with steppy faster and more effective.',
       long_description=long_description,
       url='https://github.com/minerva-ml/steppy-toolkit',
-      download_url='https://github.com/minerva-ml/steppy-toolkit/archive/0.1.11.tar.gz',
+      download_url='https://github.com/minerva-ml/steppy-toolkit/archive/0.1.12.tar.gz',
       author='Kamil A. Kaczmarek, Jakub Czakon',
       author_email='kamil.kaczmarek@neptune.ml, jakub.czakon@neptune.ml',
       keywords=['machine-learning', 'reproducibility', 'pipeline', 'tools'],
diff --git a/toolkit/lightgbm_transformers/models.py b/toolkit/lightgbm_transformers/models.py
index 4b5e400..6570440 100644
--- a/toolkit/lightgbm_transformers/models.py
+++ b/toolkit/lightgbm_transformers/models.py
@@ -69,7 +69,7 @@ def fit(self, X, y, X_valid, y_valid):
                                    **self.training_parameters)
         return self
 
-    def transform(self, X, y=None):
+    def transform(self, X, y=None, **kwargs):
         prediction = self.estimator.predict(X)
         return {'prediction': prediction}