diff --git a/Rakefile b/Rakefile
index e340041e..3b77b68d 100644
--- a/Rakefile
+++ b/Rakefile
@@ -5,7 +5,7 @@ require File.expand_path('../config/application', __FILE__)
 
 require 'rake'
 require 'rake/testtask'
-require 'rake/rdoctask'
+require 'rdoc/task'
 require 'resque/tasks'
 require 'resque_scheduler/tasks'
 
diff --git a/lib/tasks/users.rake b/lib/tasks/users.rake
index 019089fc..a89e297e 100644
--- a/lib/tasks/users.rake
+++ b/lib/tasks/users.rake
@@ -2,13 +2,13 @@ namespace :users do
   namespace :image_mode do
 
     desc "Grants given username access to image mode"
-    task :add, :username, :needs => :environment do |t, args|
+    task :add, [:username] => :environment do |t, args|
       user = toggle_user_image_mode(args, true)
       puts "#{user.name} has gained access to image mode"
     end
 
     desc "Removes given username's access to image mode"
-    task :remove, :username, :needs => :environment do |t, args|
+    task :remove, [:username] => :environment do |t, args|
       user = toggle_user_image_mode(args, false)
       puts "#{user.name} has lost access to image mode"
     end