Add instance methods to collect transport modes in StifTransportMode module

1 files changed, 1 insertions, 1 deletions

diff --git a/spec/models/line_referential_spec.rb b/spec/models/line_referential_spec.rb
index 292ecd497..07725d9e6 100644
--- a/spec/models/line_referential_spec.rb
+++ b/spec/models/line_referential_spec.rb
@@ -12,7 +12,7 @@ RSpec.describe LineReferential, :type => :model do
 
   describe "#transport_modes" do
     it 'returns a list of all transport modes' do
-      expect(FactoryGirl.create(:line_referential).transport_mode).to match_array(StifTransportModeEnumerations.transport_mode )
+      expect(FactoryGirl.create(:line_referential).transport_modes).to match_array(StifTransportModeEnumerations.transport_mode.options.map(&:first) )
     end
   end
 end