Fixed EntityFramework exception I hope.