diff --git a/elfi/methods/inference/samplers.py b/elfi/methods/inference/samplers.py
index 22c0af96..e09ba267 100644
--- a/elfi/methods/inference/samplers.py
+++ b/elfi/methods/inference/samplers.py
@@ -80,7 +80,7 @@ def __init__(self, model, discrepancy_name=None, output_names=None, **kwargs):
             Additional outputs from the model to be included in the inference result, e.g.
             corresponding summaries to the acquired samples
         kwargs:
-            See InferenceMethod
+            See ParameterInference
 
         """
         model, discrepancy_name = self._resolve_model(model, discrepancy_name)
@@ -332,7 +332,7 @@ def __init__(self, model, discrepancy_name=None, output_names=None, **kwargs):
             Additional outputs from the model to be included in the inference result, e.g.
             corresponding summaries to the acquired samples
         kwargs:
-            See InferenceMethod
+            See ParameterInference
 
         """
         model, discrepancy_name = self._resolve_model(model, discrepancy_name)
@@ -586,7 +586,7 @@ def __init__(self, model, discrepancy_name=None, output_names=None, **kwargs):
             Additional outputs from the model to be included in the inference result, e.g.
             corresponding summaries to the acquired samples
         kwargs:
-            See InferenceMethod
+            See ParameterInference
 
         """
         model, discrepancy_name = self._resolve_model(model, discrepancy_name)
@@ -695,7 +695,7 @@ def __init__(self,
         densratio_estimation : DensityRatioEstimation, optional
             Density ratio estimation object defining parameters for KLIEP
         kwargs:
-            See InferenceMethod
+            See ParameterInference
 
         """
         model, discrepancy_name = self._resolve_model(model, discrepancy_name)
diff --git a/tests/unit/test_elfi_model.py b/tests/unit/test_elfi_model.py
index 6ecb2f14..3556c415 100644
--- a/tests/unit/test_elfi_model.py
+++ b/tests/unit/test_elfi_model.py
@@ -88,7 +88,7 @@ def test_save_load(self, ma2):
 class TestNodeReference:
     def test_name_argument(self):
         # This is important because it is used when passing NodeReferences as
-        # InferenceMethod arguments
+        # ParameterInference arguments
         em.set_default_model()
         ref = em.NodeReference(name='test')
         assert str(ref) == 'test'