diff --git a/spec/models/scheme_spec.rb b/spec/models/scheme_spec.rb
index eaa11883e..1581bb1e0 100644
--- a/spec/models/scheme_spec.rb
+++ b/spec/models/scheme_spec.rb
@@ -14,7 +14,7 @@ RSpec.describe Scheme, type: :model do
       let!(:location) { FactoryBot.create(:location, scheme: scheme_1) }
       let!(:location_2) { FactoryBot.create(:location, scheme: scheme_2) }
 
-      context "when searching by code" do
+      context "when filtering by id" do
         it "returns case insensitive matching records" do
           expect(described_class.filter_by_id(scheme_1.id.to_s).count).to eq(1)
           expect(described_class.filter_by_id(scheme_1.id.to_s).first.id).to eq(scheme_1.id)
@@ -47,12 +47,14 @@ RSpec.describe Scheme, type: :model do
 
       context "when searching by all searchable fields" do
         it "returns case insensitive matching records" do
-          expect(described_class.search_by(scheme_1.code.upcase).count).to eq(1)
-          expect(described_class.search_by(scheme_1.code.downcase).count).to eq(1)
-          expect(described_class.search_by(scheme_1.code.downcase).first.code).to eq(scheme_1.code)
-          expect(described_class.search_by_service_name(scheme_2.service_name.upcase).count).to eq(1)
-          expect(described_class.search_by_service_name(scheme_2.service_name.downcase).count).to eq(1)
-          expect(described_class.search_by_service_name(scheme_2.service_name.downcase).first.service_name).to eq(scheme_2.service_name)
+          expect(described_class.search_by(scheme_1.id.to_s).count).to eq(1)
+          expect(described_class.search_by(scheme_1.id.to_s).first.id).to eq(scheme_1.id)
+          expect(described_class.search_by(scheme_2.service_name.upcase).count).to eq(1)
+          expect(described_class.search_by(scheme_2.service_name.downcase).count).to eq(1)
+          expect(described_class.search_by(scheme_2.service_name.downcase).first.service_name).to eq(scheme_2.service_name)
+          expect(described_class.search_by(location.postcode.upcase).count).to eq(1)
+          expect(described_class.search_by(location.postcode.downcase).count).to eq(1)
+          expect(described_class.search_by(location.postcode.downcase).first.locations.first.postcode).to eq(location.postcode)
         end
       end
     end