Entity helper fix