Fix entity framework beta