diff --git a/snippets/differenceBy.md b/snippets/differenceBy.md
index 2168f5adc..83b3b869c 100644
--- a/snippets/differenceBy.md
+++ b/snippets/differenceBy.md
@@ -6,7 +6,7 @@ Create a `Set` by applying `fn` to each element in `b`, then use `Array.prototyp
 
 ```js
 const differenceBy = (a, b, fn) => {
-  const s = new Set(b.map(v => fn(v)));
+  const s = new Set(b.map(fn));
   return a.filter(x => !s.has(fn(x)));
 };
 ```
diff --git a/test/differenceBy/differenceBy.js b/test/differenceBy/differenceBy.js
index e777d310c..30f251e50 100644
--- a/test/differenceBy/differenceBy.js
+++ b/test/differenceBy/differenceBy.js
@@ -1,5 +1,5 @@
 const differenceBy = (a, b, fn) => {
-  const s = new Set(b.map(v => fn(v)));
+  const s = new Set(b.map(fn));
   return a.filter(x => !s.has(fn(x)));
 };
 module.exports = differenceBy;