Update migration example

To match the current implementation