Atualização do Entity Framework 5 para 6

Depois de atualizar nosso projeto do Entity Framework 5 para o Entity Framework 6 (apesar da função de atualização do NuGets), recebo o seguinte erro na minha classe Entities gerada:

Erro 1 O tipo ou nome do espaço para nome 'Objetos' não existe no espaço para nome 'System.Data'
(está faltando uma referência de montagem?)

Entendo que isso ocorre porque o espaço para nome foi alterado e posso corrigir manualmente o erro alterando minhas importações de:using System.Data.Objects; eusing System.Data.Objects.DataClasses; Para:using System.Data.Entity.Core.Objects;

No entanto, o arquivo é gerado, então eu preciso reaplicar essa correção após cadaUpdate model from Database. Existe algo extra a ser alterado para que o EF gere o modelo sem esse erro.