fixed entity.py