Added post_migrate hook