Fix test migrations