tasks/role.rb in gaptool-server-0.6.6 vs tasks/role.rb in gaptool-server-0.7.0

@@ -1,9 +1,9 @@
 namespace :role do
   desc "Set security group for a role"
   task :set_sg, [:role, :environment, :security_group] do |t, args|
     abort("Missing parameters") if args[:role].nil? || args[:environment].nil? || args[:role].empty? || args[:environment].empty? || args[:security_group].nil? || args[:security_group].empty?
-    role = DH.get_role_data(args[:role])
+    role = DH.get_role_data(args[:role], args[:environment])
     role['sg'].merge!(args[:environment] => args[:security_group])
     DH.save_role_data(args[:role], role)
   end
 end