Fixing compilation issue with EntiryFramework