diff --git a/rektgbm/cli.py b/rektgbm/cli.py
index 504a898..257b845 100644
--- a/rektgbm/cli.py
+++ b/rektgbm/cli.py
@@ -46,7 +46,7 @@ def main(
         typer.Argument(help="Path to save the prediction results."),
     ] = "predict.csv",
     n_trials: Annotated[
-        int, typer.Argument(help="Number of optimization trials.")
+        int, typer.Argument(help="Number of optimization trials per model.")
     ] = 100,
 ) -> None:
     _supported_type = (CSV, PARQUET)
diff --git a/rektgbm/encoder.py b/rektgbm/encoder.py
index cc3ca10..198c2e4 100644
--- a/rektgbm/encoder.py
+++ b/rektgbm/encoder.py
@@ -34,5 +34,5 @@ def fit_transform_label(self, series: YdataLike) -> YdataLike:
         self.fit_label(series=series)
         return self.transform_label(series=series)
 
-    def inverse_transform(self, series: XdataLike) -> XdataLike:
+    def inverse_transform(self, series: YdataLike) -> YdataLike:
         return self.label_encoder.inverse_transform(y=series)