lib/vagrant_dns/middleware/update.rb in vagrant-dns-server-0.3 vs lib/vagrant_dns/middleware/update.rb in vagrant-dns-server-0.4

@@ -6,10 +6,10 @@
 	  @app = app
 	end
 
 	def call(env)
 	  @app.call(env)
-	  update env[:vm] if env["vm"].state == :running
+	  update env[:machine] if env[:machine].provider.state.id == :running
 	end
 
 	protected
 	def update(vm)
         VagrantDns::NetworkBinder.new.bind(vm)