Use EntityDocument instead of Entity