lib/code/parser/if.rb in template-ruby-0.2.4 vs lib/code/parser/if.rb in template-ruby-0.3.0

@@ -1,10 +1,10 @@
 class Code
   class Parser
     class If < Parslet::Parser
       rule(:if_modifier) { ::Code::Parser::IfModifier.new }
-      rule(:code) { ::Code::Parser::Code.new }
+      rule(:code) { ::Code::Parser::Code.new.present }
 
       rule(:if_keyword) { str("if") }
       rule(:else_keyword) { str("else") }
       rule(:unless_keyword) { str("unless") }
       rule(:end_keyword) { str("end") }
@@ -14,16 +14,16 @@
       rule(:whitespace) { (space | newline).repeat(1) }
 
       rule(:if_rule) do
         (
           (if_keyword | unless_keyword).as(:if_operator) >> whitespace >>
-            if_modifier.as(:if_statement) >> code.as(:if_body) >>
+          if_modifier.as(:if_statement) >> code.as(:if_body).maybe >>
             (
               else_keyword >>
                 (
                   whitespace >> (if_keyword | unless_keyword).as(:operator) >>
                     whitespace >> if_modifier.as(:statement)
-                ).maybe >> code.as(:body)
+                ).maybe >> code.as(:body).maybe
             ).repeat(1).as(:elses).maybe >> end_keyword
         ).as(:if) | if_modifier
       end
 
       root(:if_rule)