service_messages_spec.rb in rake-funnel-0.21.1

- old
+ new

@@ -1,11 +1,9 @@
 # rubocop:disable RSpec/FilePath
 
-include Rake::Funnel::Integration
-
 describe Rake::Funnel::Integration::TeamCity::ServiceMessages do
   before do
-    allow(TeamCity).to receive(:running?).and_return(teamcity_running?)
+    allow(Rake::Funnel::Integration::TeamCity).to receive(:running?).and_return(teamcity_running?)
     allow($stdout).to receive(:puts)
   end
 
   context 'when running outside TeamCity' do
     let(:teamcity_running?) { false }