Change entity classes