Fixed typehint
Merge pull request #145 from CoBase/entity-fix

Fixed typehint