Rakefile in openstudio-extension-0.2.1 vs Rakefile in openstudio-extension-0.2.2

@@ -34,10 +34,10 @@
 RSpec::Core::RakeTask.new(:spec)
 
 require 'openstudio/extension/rake_task'
 require 'openstudio/extension'
 rake_task = OpenStudio::Extension::RakeTask.new
-rake_task.set_extension_class(OpenStudio::Extension::Extension)
+rake_task.set_extension_class(OpenStudio::Extension::Extension, 'nrel/openstudio-extension-gem')
 
 require 'rubocop/rake_task'
 RuboCop::RakeTask.new
 
 task default: :spec