1 files changed, 5 insertions, 2 deletions
diff --git a/spec/lib/constraints/feature_constrainer_spec.rb b/spec/lib/constraints/feature_constrainer_spec.rb
index 0739da801a7..7665d5b3547 100644
--- a/spec/lib/constraints/feature_constrainer_spec.rb
+++ b/spec/lib/constraints/feature_constrainer_spec.rb
@@ -5,9 +5,12 @@ require 'spec_helper'
 describe Constraints::FeatureConstrainer do
   describe '#matches' do
     it 'calls Feature.enabled? with the correct arguments' do
-      expect(Feature).to receive(:enabled?).with(:feature_name, "an object", default_enabled: true)
+      gate = stub_feature_flag_gate("an object")
 
-      described_class.new(:feature_name, "an object", default_enabled: true).matches?(double('request'))
+      expect(Feature).to receive(:enabled?)
+        .with(:feature_name, gate, default_enabled: true)
+
+      described_class.new(:feature_name, gate, default_enabled: true).matches?(double('request'))
     end
   end
 end