Build fix relating to migrations