Add type to extended entity