minor fix in migtration
migration and models update