Added Entity annotation
fix