fix entity bug