diff --git a/graphql-jpa-query-example-model-starwars/src/main/java/com/introproventures/graphql/jpa/query/schema/model/starwars/DroidFunction.java b/graphql-jpa-query-example-model-starwars/src/main/java/com/introproventures/graphql/jpa/query/schema/model/starwars/DroidFunction.java
index 56bcefa49..1513bc901 100644
--- a/graphql-jpa-query-example-model-starwars/src/main/java/com/introproventures/graphql/jpa/query/schema/model/starwars/DroidFunction.java
+++ b/graphql-jpa-query-example-model-starwars/src/main/java/com/introproventures/graphql/jpa/query/schema/model/starwars/DroidFunction.java
@@ -1,14 +1,16 @@
 package com.introproventures.graphql.jpa.query.schema.model.starwars;
 
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import javax.persistence.Table;
+
 import com.introproventures.graphql.jpa.query.annotation.GraphQLDescription;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 
-import javax.persistence.Entity;
-import javax.persistence.Id;
-
 
-@Entity(name = "droid_function")
+@Entity
+@Table(name = "droid_function")
 @GraphQLDescription("Represents the functions a droid can have")
 @Data
 @EqualsAndHashCode()