diff --git a/spring-data-rest-webmvc/src/test/groovy/org/springframework/data/rest/webmvc/spec/RepositoryRestControllerSpec.groovy b/spring-data-rest-webmvc/src/test/groovy/org/springframework/data/rest/webmvc/spec/RepositoryRestControllerSpec.groovy
index f91cdd9cf..275534c8e 100644
--- a/spring-data-rest-webmvc/src/test/groovy/org/springframework/data/rest/webmvc/spec/RepositoryRestControllerSpec.groovy
+++ b/spring-data-rest-webmvc/src/test/groovy/org/springframework/data/rest/webmvc/spec/RepositoryRestControllerSpec.groovy
@@ -86,7 +86,7 @@ class RepositoryRestControllerSpec extends Specification {
 
     then:
     mv.model.status == HttpStatus.OK
-    reposLinks?.size() == 3
+    reposLinks?.size() == 4
 
     when: "adding an entity"
     model.clear()
diff --git a/spring-data-rest-webmvc/src/test/resources/META-INF/persistence.xml b/spring-data-rest-webmvc/src/test/resources/META-INF/persistence.xml
index 87f693d94..57a41a4cf 100644
--- a/spring-data-rest-webmvc/src/test/resources/META-INF/persistence.xml
+++ b/spring-data-rest-webmvc/src/test/resources/META-INF/persistence.xml
@@ -1,6 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <persistence xmlns="http://java.sun.com/xml/ns/persistence" version="2.0">
   <persistence-unit name="jpa.sample">
+    <class>org.springframework.data.rest.test.webmvc.Family</class>
     <class>org.springframework.data.rest.test.webmvc.Person</class>
     <class>org.springframework.data.rest.test.webmvc.Profile</class>
     <class>org.springframework.data.rest.test.webmvc.Address</class>