Correct for broken integration test
Add type to entity map