diff --git a/demo-scala/src/UnweightedStream.scala b/demo-scala/src/UnweightedStream.scala
index 7cfb2d2..5d6aee2 100644
--- a/demo-scala/src/UnweightedStream.scala
+++ b/demo-scala/src/UnweightedStream.scala
@@ -20,7 +20,7 @@ class SamplingTraversableOnce[T](val it: TraversableOnce[T]) {
     }
 
   /**
-    * Samples this iterator using the provided algorithm and returns a copy of the reservoir.
+    * Samples this [[TraversableOnce]] using the provided algorithm and returns a copy of the reservoir.
     *
     * @param rs the sampling algorithm
     * @return a [[List]] containing the sampled elements
diff --git a/demo-scala/src/WeightedStream.scala b/demo-scala/src/WeightedStream.scala
index 0299210..45272f4 100644
--- a/demo-scala/src/WeightedStream.scala
+++ b/demo-scala/src/WeightedStream.scala
@@ -20,7 +20,7 @@ class WeightedSamplingTraversableOnce[T](val it: TraversableOnce[(T, Double)]) {
     }
 
   /**
-    * Samples this iterator using the provided algorithm and returns a copy of the reservoir.
+    * Samples this [[TraversableOnce]] using the provided algorithm and returns a copy of the reservoir.
     *
     * @param wrs the sampling algorithm
     * @return a [[List]] containing the sampled elements