lib/fog/azurerm/models/network/network_security_groups.rb in fog-azure-rm-0.1.0 vs lib/fog/azurerm/models/network/network_security_groups.rb in fog-azure-rm-0.1.1

@@ -1,25 +1,25 @@
 module Fog
   module Network
     class AzureRM
       # collection class for Network Security Group
       class NetworkSecurityGroups < Fog::Collection
-        model NetworkSecurityGroup
+        model Fog::Network::AzureRM::NetworkSecurityGroup
         attribute :resource_group
 
         def all
           requires :resource_group
           network_security_groups = []
           service.list_network_security_groups(resource_group).each do |nsg|
-            network_security_groups << NetworkSecurityGroup.parse(nsg)
+            network_security_groups << Fog::Network::AzureRM::NetworkSecurityGroup.parse(nsg)
           end
           load(network_security_groups)
         end
 
         def get(resource_group, name)
-          nsg = service.get_network_security_group(resource_group, name)
-          network_security_group = NetworkSecurityGroup.new(service: service)
-          network_security_group.merge_attributes(NetworkSecurityGroup.parse(nsg))
+          network_security_group = service.get_network_security_group(resource_group, name)
+          network_security_group_fog = Fog::Network::AzureRM::NetworkSecurityGroup.new(service: service)
+          network_security_group_fog.merge_attributes(Fog::Network::AzureRM::NetworkSecurityGroup.parse(network_security_group))
         end
       end
     end
   end
 end