Fix documentation for Entity