lib/rake/funnel/integration/teamcity/nunit_plugin.rb in rake-funnel-0.3.0.pre vs lib/rake/funnel/integration/teamcity/nunit_plugin.rb in rake-funnel-0.3.1.pre

@@ -1,7 +1,9 @@
 module Rake::Funnel::Integration::TeamCity
   class NUnitPlugin
+    include Rake::Funnel::Support
+
     ENV_VAR = 'teamcity.dotnet.nunitaddin'
 
     class << self
       def setup(nunit_executable)
         addins = get_addin_dir(ENV[ENV_VAR]) || return
@@ -18,10 +20,10 @@
 
         File.expand_path(source)
       end
 
       def read_version(executable)
-        nunit = Rake::Funnel::Support::Which.which(executable) || return
+        nunit = Which.which(executable) || return
         version = BinaryVersionReader.read_from(nunit)
 
         unless version.file_version
           Rake.rake_output_message("Could read version from NUnit executable in #{nunit}")
           return