diff --git a/spec/features/schemes_spec.rb b/spec/features/schemes_spec.rb
index 5ebffdc42..8f4c755ed 100644
--- a/spec/features/schemes_spec.rb
+++ b/spec/features/schemes_spec.rb
@@ -128,7 +128,7 @@ RSpec.describe "Schemes scheme Features" do
 
               it "displays all schemes after I clear the search results" do
                 click_link("Clear search")
-                expect(page).to have_content(scheme_to_search.code)
+                expect(page).to have_content(scheme_to_search.id_to_display)
                 schemes.each do |scheme|
                   expect(page).to have_content(scheme.id)
                 end
@@ -160,7 +160,7 @@ RSpec.describe "Schemes scheme Features" do
           end
 
           it "shows me details about the selected scheme" do
-            expect(page).to have_content(schemes.first.code)
+            expect(page).to have_content(schemes.first.id_to_display)
             expect(page).to have_content(schemes.first.service_name)
             expect(page).to have_content(schemes.first.sensitive)
             expect(page).to have_content(schemes.first.scheme_type)
diff --git a/spec/requests/schemes_controller_spec.rb b/spec/requests/schemes_controller_spec.rb
index 682332011..922e7e01f 100644
--- a/spec/requests/schemes_controller_spec.rb
+++ b/spec/requests/schemes_controller_spec.rb
@@ -551,7 +551,7 @@ RSpec.describe SchemesController, type: :request do
         expect(Scheme.last.secondary_client_group).to eq(nil)
         expect(Scheme.last.support_type).to eq(nil)
         expect(Scheme.last.intended_stay).to eq(nil)
-        expect(Scheme.last.code).to match(/S*/)
+        expect(Scheme.last.id_to_display).to match(/S*/)
       end
     end
 
@@ -585,7 +585,7 @@ RSpec.describe SchemesController, type: :request do
         expect(Scheme.last.secondary_client_group).to eq(nil)
         expect(Scheme.last.support_type).to eq(nil)
         expect(Scheme.last.intended_stay).to eq(nil)
-        expect(Scheme.last.code).to match(/S*/)
+        expect(Scheme.last.id_to_display).to match(/S*/)
       end
 
       context "when required organisation id param is missing" do