Hm, we need a better way to do this rather than putting this code in every entity!
But let's come back to that after this pull request.