diff --git a/test/ui_test.rb b/test/ui_test.rb
index 42d44a209..dca5a46d7 100644
--- a/test/ui_test.rb
+++ b/test/ui_test.rb
@@ -29,7 +29,7 @@ class ExampleTest < Test::Unit::TestCase
        element_password.send_keys "123456"
        element_password.submit
        wait = Selenium::WebDriver::Wait.new(:timeout => 10) # seconds
-       wait.until { driver.find_element(:id => "app") }
+       wait.until { browser.find_element(:id => "app") }
        assert_equal browser.current_url, "https://portal.znuny.com/#login"
     end
 
@@ -41,7 +41,7 @@ class ExampleTest < Test::Unit::TestCase
        element_password.send_keys "090504"
        element_password.submit
        wait = Selenium::WebDriver::Wait.new(:timeout => 10) # seconds
-       wait.until { driver.find_element(:id => "app") }
+       wait.until { browser.find_element(:id => "app") }
        assert_equal browser.current_url, "https://portal.znuny.com/#ticket_view/my_tickets"
     end
 end