Fix EntityFramework reference