diff --git a/features/steps/user.rb b/features/steps/user.rb
index b1d088f07f992e6be6b3f9992a924a0a8f82bc62..59385a6ab59d03ec4552f9449f75fc048e2528ca 100644
--- a/features/steps/user.rb
+++ b/features/steps/user.rb
@@ -34,7 +34,7 @@ class Spinach::Features::User < Spinach::FeatureSteps
   end
 
   step 'I should see contributions calendar' do
-    expect(page).to have_css('.cal-heatmap-container')
+    expect(page).to have_css('.js-contrib-calendar')
   end
 
   def contributed_project