test/models/compute/test_availability_set.rb in fog-azure-rm-0.0.4 vs test/models/compute/test_availability_set.rb in fog-azure-rm-0.0.5

@@ -3,10 +3,11 @@
 # Test class for Availability Set Model
 class TestAvailabilitySet < Minitest::Test
   def setup
     @service = Fog::Compute::AzureRM.new(credentials)
     @availability_set = availability_set(@service)
-    @response = ApiStub::Models::Compute::AvailabilitySet.create_availability_set_response
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @response = ApiStub::Models::Compute::AvailabilitySet.create_availability_set_response(compute_client)
   end
 
   def test_model_methods
     methods = [
       :save,