Fix mismatched doc parameter and entity type error