lib/fog/azurerm/models/compute/availability_sets.rb in fog-azure-rm-0.0.4 vs lib/fog/azurerm/models/compute/availability_sets.rb in fog-azure-rm-0.0.5

@@ -21,10 +21,12 @@
           end
           load(accounts)
         end
 
         def get(resource_group, identity)
-          all.find { |as| as.resource_group == resource_group && as.name == identity }
+          availability_set = service.get_availability_set(resource_group, identity)
+          availability_set_obj = Fog::Compute::AzureRM::AvailabilitySet.new(service: service)
+          availability_set_obj.merge_attributes(Fog::Compute::AzureRM::AvailabilitySet.parse(availability_set))
         end
       end
     end
   end
 end