diff --git a/lib/foreman_discovery.rake b/lib/foreman_discovery.rake
index 06bc9ebf..051555d1 100644
--- a/lib/foreman_discovery.rake
+++ b/lib/foreman_discovery.rake
@@ -4,11 +4,7 @@ namespace :test do
   Rake::TestTask.new(:foreman_discovery) do |t|
     test_dir = File.join(File.dirname(__FILE__), '..', 'test')
     t.libs << ["test",test_dir]
-    if ENV['FILE'] || ENV['TEST']
-      t.pattern = ENV['FILE'] || ENV['TEST']
-    else
-      t.pattern = "#{test_dir}/**/*_test.rb"
-    end
+    t.pattern = ENV['FILE'] || ENV['TEST'] || "#{test_dir}/**/*_test.rb"
     t.verbose = true
     t.warning = false
   end