diff --git a/spec/jam_ruby/connection_manager_spec.rb b/spec/jam_ruby/connection_manager_spec.rb
index 0a4b45a4c..7a17ce8ab 100644
--- a/spec/jam_ruby/connection_manager_spec.rb
+++ b/spec/jam_ruby/connection_manager_spec.rb
@@ -407,7 +407,7 @@ describe ConnectionManager do
     user = User.find(user_id)
     music_session = MusicSession.find(music_session_id)
 
-    @connman.create_connection(user_id, client_id, "1.1.1.1")
+    connection = @connman.create_connection(user_id, client_id, "1.1.1.1")
     @connman.join_music_session(user, client_id, music_session, true, TRACKS)
 
     assert_session_exists(music_session_id, true)
@@ -416,7 +416,7 @@ describe ConnectionManager do
       result.getvalue(0, 0).should == music_session_id
     end
 
-    @connman.leave_music_session(user, client_id, music_session)
+    @connman.leave_music_session(user, connection, music_session)
 
     @conn.exec("SELECT music_session_id FROM connections WHERE client_id = $1", [client_id]) do |result|
       result.getvalue(0, 0).should == nil