Using Entity Framework Power Tools Beta 2 extension, it generates model classes with the following imports:
using System.Data.Entity.Infrastructure;
Don't add EF lib via "Add reference" dialog (it's old version) You need to add NuGet package "EntityFramework" of latest version.