From b8939a17bf677a7580de6e60e88e9e97ec1b4a02 Mon Sep 17 00:00:00 2001
From: Fabio Yamate <fabio.yamate@plataformatec.com.br>
Date: Tue, 27 Sep 2011 00:59:06 -0300
Subject: [PATCH] Small clean up.

---
 lib/simple_form.rb                | 6 +-----
 test/form_builder/wrapper_test.rb | 1 -
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/lib/simple_form.rb b/lib/simple_form.rb
index 55e97009..59b5a0ed 100644
--- a/lib/simple_form.rb
+++ b/lib/simple_form.rb
@@ -117,11 +117,7 @@ module SimpleForm
 
   # Retrieves a given wrapper
   def self.wrapper(name)
-    if wrapper = @@wrappers[name.to_sym]
-      wrapper
-    else
-      raise WrapperNotFound, "Couldn't find wrapper with name #{name}"
-    end
+    @@wrappers[name.to_sym] or raise WrapperNotFound, "Couldn't find wrapper with name #{name}"
   end
 
   # Raised when fails to find a given wrapper name
diff --git a/test/form_builder/wrapper_test.rb b/test/form_builder/wrapper_test.rb
index 6932b892..142883ad 100644
--- a/test/form_builder/wrapper_test.rb
+++ b/test/form_builder/wrapper_test.rb
@@ -118,7 +118,6 @@ class WrapperTest < ActionView::TestCase
       with_form_for @user, :name, :wrapper => "another"
       assert_select "section.custom_wrapper div.another_wrapper label"
       assert_select "section.custom_wrapper div.another_wrapper input.string"
-      output_buffer.replace ""
     end
   end