Revert "Fixed error in entity"

This reverts commit 44a3e5a937ea3631f006890d115e1ab29439882b.
Revert