Make migration tool able to override data model