diff --git a/libs/wlgen/wlgen/workload.py b/libs/wlgen/wlgen/workload.py
index 66903f71fc63111ea24f8dd66f44c26b9c30d66b..6e65ca9cb1144aa5b799aca9bd3bc374e178bf56 100644
--- a/libs/wlgen/wlgen/workload.py
+++ b/libs/wlgen/wlgen/workload.py
@@ -165,8 +165,9 @@ class Workload(object):
         This method starts the execution of the workload. If the user provides
         an ftrace object, the method will also collect a trace.
 
-        :param ftrace: FTrace object to collect a trace
-        :type ftrace: :mod:`trappy.trace.FTrace`
+        :param ftrace: FTrace object to collect a trace. If using
+            :class:`TestEnv`, you can use :attr:`TestEnv.ftrace` for this.
+        :type ftrace: :mod:`devlib.trace.FTraceCollector`
 
         :param cgroup: specifies the cgroup name in which the workload has to
                        run