Can you send a PR for this migration too?  It's helpful!