tasks/benchmark_get_treatment.rake in splitclient-rb-1.0.4 vs tasks/benchmark_get_treatment.rake in splitclient-rb-2.0.0

@@ -26,10 +26,10 @@
   user_id = ENV['user_id'].nil? ? "fake_id_1" : ENV['user_id']
   feature_id = ENV['feature_id'].nil? ? "sample_feature" : ENV['feature_id']
 
   threads = []
   times_per_thread = iterations / 4
-  split_client = SplitIoClient::SplitClient.new(api_key, {base_uri: base_uri, logger: Logger.new("/dev/null") })
+  split_client = SplitIoClient::SplitFactory.new(api_key, {base_uri: base_uri, logger: Logger.new("/dev/null").client })
 
   puts Benchmark.measure{
     4.times do |i|
       threads << Thread.new do
         times_per_thread.times do