Bug fix in DB migrations