Fixed nil sections issue

spinatelli · Jun 16, 2013 · 981dabb · 981dabb
1 parent 6d7c8cb
commit 981dabb
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/rlatex.rb b/rlatex.rb
@@ -43,19 +43,15 @@ def create_main_tex()
       f.write("\\begin{document}\n\n")
       f.write("\\maketitle\n")
       f.write("  %%% INPUT\n")
-      @sections.each do |section|
-        f.write("  \\input{contents/#{section}.tex}\n")
-      end
+      @sections.each { |s| f.write("  \\input{contents/#{s}.tex}\n") } unless @sections.nil?
 
       f.write("\\end{document}\n")
       f.write("\n")
     end
   end
 
   def create_sections()
-    @sections.each do |section|
-      create_section section
-    end
+    @sections.each { |section| create_section section } unless @sections.nil?
   end
 
   def create_section(section)