diff --git a/lib/cyberarm_engine/ui/dsl.rb b/lib/cyberarm_engine/ui/dsl.rb
index 579a236..2503eaf 100644
--- a/lib/cyberarm_engine/ui/dsl.rb
+++ b/lib/cyberarm_engine/ui/dsl.rb
@@ -70,16 +70,14 @@ module CyberarmEngine
       options[:parent] = element_parent
       options[:theme] = current_theme
 
-      element_parent( Element::Progress.new(options, block) )
+      add_element( Element::Progress.new(options, block) )
     end
 
     def background(color = Gosu::Color::NONE)
-      @containers.last.style.background = color
+      element_parent.style.background = color
     end
 
     def theme(theme)
-      self.is_a?(CyberarmEngine::Element::Container) ? self : @containers.last
-
       element_parent.options[:theme] = theme
     end
 
diff --git a/lib/cyberarm_engine/ui/elements/container.rb b/lib/cyberarm_engine/ui/elements/container.rb
index 334883b..2bf454c 100644
--- a/lib/cyberarm_engine/ui/elements/container.rb
+++ b/lib/cyberarm_engine/ui/elements/container.rb
@@ -169,4 +169,4 @@ module CyberarmEngine
       end
     end
   end
-end
\ No newline at end of file
+end