1 files changed, 3 insertions, 3 deletions
diff --git a/spec/lib/gitlab/cycle_analytics/shared_stage_spec.rb b/spec/lib/gitlab/cycle_analytics/shared_stage_spec.rb
index 6f3883d80f8..08425acbfc8 100644
--- a/spec/lib/gitlab/cycle_analytics/shared_stage_spec.rb
+++ b/spec/lib/gitlab/cycle_analytics/shared_stage_spec.rb
@@ -9,15 +9,15 @@ shared_examples 'base stage' do
   end
 
   it 'has the median data value' do
-    expect(stage.median_data[:value]).not_to be_nil
+    expect(stage.as_json[:value]).not_to be_nil
   end
 
   it 'has the median data stage' do
-    expect(stage.median_data[:title]).not_to be_nil
+    expect(stage.as_json[:title]).not_to be_nil
   end
 
   it 'has the median data description' do
-    expect(stage.median_data[:description]).not_to be_nil
+    expect(stage.as_json[:description]).not_to be_nil
   end
 
   it 'has the title' do