diff --git a/examples/BookStore.php b/examples/BookStore.php
index c0a0613..2d1b33d 100644
--- a/examples/BookStore.php
+++ b/examples/BookStore.php
@@ -14,11 +14,11 @@ class BookStore extends GDS\Store
      */
     protected function buildSchema()
     {
-        $this->setEntityClass('\\Book');
         return (new GDS\Schema('Book'))
             ->addString('title')
             ->addString('author')
-            ->addString('isbn', TRUE);
+            ->addString('isbn', TRUE)
+            ->setEntityClass('\\Book');
     }
 
 }
\ No newline at end of file