Fix JPA bugs