lib/epubber/models/concerns/has_chapters.rb in epubber-0.0.4 vs lib/epubber/models/concerns/has_chapters.rb in epubber-0.1.0

@@ -5,12 +5,13 @@
     def chapters
       @chapters ||= []
     end
 
     # Add chapter
-    def chapter(&block)
+    def chapter
+      raise 'No block given' unless block_given?
       chapter = Epubber::Models::Chapter.new
-      chapter.instance_eval &block
+      yield chapter
       chapter.id(chapters.count + 1)
       chapters << chapter
     end
 
     def contextified_chapters