fix support for migrations
fix support for migrations