diff --git a/lib/draper/tasks/test.rake b/lib/draper/tasks/test.rake
index 03cc4f3..df54402 100644
--- a/lib/draper/tasks/test.rake
+++ b/lib/draper/tasks/test.rake
@@ -1,6 +1,6 @@
 require 'rake/testtask'
 
-test_task = if Rails.version.start_with?("3.0")
+test_task = if Rails.version.to_f < 3.2
   require 'rails/test_unit/railtie'
   Rake::TestTask
 else