diff --git a/test/client/tasks_test.rb b/test/client/tasks_test.rb
index 0fef3c6..f3f2018 100644
--- a/test/client/tasks_test.rb
+++ b/test/client/tasks_test.rb
@@ -14,7 +14,7 @@ describe Elastomer::Client::Tasks do
   end

   it "list all in-flight tasks" do
-    h = @tasks.get :group_by => "nodes"
+    h = @tasks.get
     assert h["nodes"].keys.size > 0

     total_tasks = h["nodes"].map { |k, v| v["tasks"].keys.count }.sum
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 1e6fbc1..b128faf 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -18,6 +18,7 @@ end
 require "minitest/spec"
 require "minitest/autorun"
 require "minitest/focus"
+require "active_support/core_ext/enumerable"

 # push the lib folder onto the load path
 $LOAD_PATH.unshift "lib"