diff --git a/spec/features/user_managing_broadcasts_spec.rb b/spec/features/user_managing_broadcasts_spec.rb
index 00ca6bb..44a4eea 100644
--- a/spec/features/user_managing_broadcasts_spec.rb
+++ b/spec/features/user_managing_broadcasts_spec.rb
@@ -91,7 +91,7 @@ feature 'User managing broadcasts' do
       BroadcastsChannel.broadcast_stream_updates(broadcast)
 
       expect(page).to have_content stream_begun_message
-      expect(page).not_to have_css('div#broadcast_video[video-type="stream"]')
+      expect(page).to have_selector('div#broadcast_video', count: 1)
 
       broadcast.streamer_status = :idle
       broadcast.status = :idle
@@ -101,14 +101,14 @@ feature 'User managing broadcasts' do
       click_on recording.download_file_name
 
       expect(page).to have_content stream_idle_message
-      expect(page).to have_css('div#broadcast_video[video-type="recording"]')
+      expect(page).to have_selector('div#broadcast_video', count: 1)
 
       broadcast.streamer_status = :recording
       broadcast.status = :active
       BroadcastsChannel.broadcast_stream_updates(broadcast)
 
       expect(page).to have_content stream_begun_message
-      expect(page).to have_css('div#broadcast_video[video-type="recording"]')
+      expect(page).to have_selector('div#broadcast_video', count: 1)
     end
 
     scenario 'user can go back and forth between live session and previous sessions', js: true do