diff --git a/Rakefile b/Rakefile
index ea6351f4..60e3565a 100644
--- a/Rakefile
+++ b/Rakefile
@@ -6,10 +6,11 @@ require 'pry'
 
 task :default => :test
 
-Rake::Task["test"].enhance [:check_samples]
-
 Rake::TestTask.new
 
+# Extend test task to check for samples
+task :test => :check_samples
+
 desc "Check that we have samples.json generated"
 task :check_samples do
   unless File.exist?('lib/linguist/samples.json')