fix entity function