diff --git a/batchflow/models/torch/base.py b/batchflow/models/torch/base.py
index 50781fd85..929fa4cba 100755
--- a/batchflow/models/torch/base.py
+++ b/batchflow/models/torch/base.py
@@ -1601,7 +1601,7 @@ def apply_output_operation(predictions, operation):
 
                 predictions = predictions[0] if isinstance(predictions, (tuple, list)) else predictions
                 if operation == 'softplus':
-                    result = torch.nn.functional.softplus(tensor) # pylint: disable=not-callable
+                    result = torch.nn.functional.softplus(predictions) # pylint: disable=not-callable
                 elif operation == 'sigmoid':
                     result = torch.sigmoid(predictions)
                 elif operation == 'sigmoid_uint8':