DB Migrate fix