lib/vagrant-hosts/addresses.rb in vagrant-hosts-2.1.0 vs lib/vagrant-hosts/addresses.rb in vagrant-hosts-2.1.1

@@ -3,13 +3,13 @@
   private
 
   def all_hosts(config)
 
     all_hosts = []
-    all_hosts += local_hosts
+    all_hosts += local_hosts(@machine)
 
     if config.autoconfigure
-      all_hosts += vagrant_hosts
+      all_hosts += vagrant_hosts(@env)
     end
     all_hosts += config.hosts
 
     all_hosts
   end