diff --git a/core/src/main/scala/za/co/absa/db/fadb/status/handling/StatusHandling.scala b/core/src/main/scala/za/co/absa/db/fadb/status/handling/StatusHandling.scala
index ba2519c..43e773c 100644
--- a/core/src/main/scala/za/co/absa/db/fadb/status/handling/StatusHandling.scala
+++ b/core/src/main/scala/za/co/absa/db/fadb/status/handling/StatusHandling.scala
@@ -21,15 +21,15 @@ import za.co.absa.db.fadb.status.FunctionStatus
 
 /**
  *  `StatusHandling` is a base trait that defines the interface for handling the status of a function invocation.
- *  It provides a method to check the status of a function invocation with data.
+ *  It provides a method to check the status of a function invocation.
  */
 trait StatusHandling {
 
   /**
    *  Checks the status of a function invocation.
-   *  @param statusWithData - The status of the function invocation with data.
-   *  @return Either a `StatusException` if the status code indicates an error, or the data (along with the status
-    *          information so that it's retrievable) if the status code is successful.
+   *  @param functionStatus - The status of the function invocation.
+   *  @return Some with a `StatusException` if the status code indicates an error,
+   *          or None if the status code is successful.
    */
   def checkStatus(functionStatus: FunctionStatus): Option[StatusException]
 }