diff --git a/test/integration/tc_login.rb b/test/integration/tc_login.rb
index b49041b24..15b99297c 100644
--- a/test/integration/tc_login.rb
+++ b/test/integration/tc_login.rb
@@ -12,7 +12,7 @@ class TC_Login < Test::Unit::TestCase
   include RSpec::Matchers
 
   def test_log_in
-    session = Capybara::Session.new(:selenium)
+    session = Capybara::Session.new(:selenium_headless)
     session.visit(ATTACK_URL)
     sleep 2.0
     BeefTest.save_screenshot(session)
@@ -82,7 +82,7 @@ class TC_Login < Test::Unit::TestCase
 
     attacker.should have_content('Details')
     attacker.should have_content('Commands')
-    attacker.should have_content('Rider')
+    # attacker.should have_content('Rider') # Old functionality
 
     BeefTest.save_screenshot(attacker)
     BeefTest.save_screenshot(victim)