diff --git a/README.md b/README.md
index 9ab72488..69362518 100644
--- a/README.md
+++ b/README.md
@@ -473,7 +473,7 @@ The created file will have the following content:
 
 ```js
 // here, 'subtitle' is the name of the field defined when you execute the generate command
-module.exports = function (block) {
+module.exports = function (block, slug) {
   // Example to change a string to boolean
   // block.subtitle = !!(block.subtitle)
 
diff --git a/src/tasks/templates/migration-file.js b/src/tasks/templates/migration-file.js
index 44b2cdf0..94304a9d 100644
--- a/src/tasks/templates/migration-file.js
+++ b/src/tasks/templates/migration-file.js
@@ -1,4 +1,4 @@
-export default `export default function (block) {
+export default `export default function (block, slug) {
   // Example to change a string to boolean
   // block.{{ fieldname }} = !!(block.{{ fieldname }})