diff --git a/bilby/core/sampler/__init__.py b/bilby/core/sampler/__init__.py
index 548ddb9448111c253d3035718b448be7a60a4e6c..7ec87be2335142070891dd3276b4c64dbbec5b3e 100644
--- a/bilby/core/sampler/__init__.py
+++ b/bilby/core/sampler/__init__.py
@@ -3,6 +3,7 @@ import sys
 import datetime
 from collections import OrderedDict
 
+import bilby
 from ..utils import command_line_args, logger
 from ..prior import PriorDict, DeltaFunction
 
@@ -107,7 +108,7 @@ def run_sampler(likelihood, priors=None, label='label', outdir='outdir',
 
     Returns
     -------
-    result
+    result: bilby.core.result.Result
         An object containing the results
     """