Revert "Update entity.py"

This reverts commit f647117e3872ca5b1ed5b19a80d43f39cb68ea07.