diff --git a/test/unit/plugins/providers/docker/action/connect_networks_test.rb b/test/unit/plugins/providers/docker/action/connect_networks_test.rb
index 54827ff4a..78e2a7d34 100644
--- a/test/unit/plugins/providers/docker/action/connect_networks_test.rb
+++ b/test/unit/plugins/providers/docker/action/connect_networks_test.rb
@@ -23,6 +23,7 @@ describe VagrantPlugins::DockerProvider::Action::ConnectNetworks do
 
   let(:machine) do
     iso_env.machine(iso_env.machine_names[0], :docker).tap do |m|
+      allow(m).to receive(:vagrantfile).and_return(vagrantfile)
       allow(m).to receive(:config).and_return(machine_config)
       allow(m).to receive(:id).and_return("12345")
       allow(m.provider).to receive(:driver).and_return(driver)
@@ -65,10 +66,6 @@ describe VagrantPlugins::DockerProvider::Action::ConnectNetworks do
 
   subject { described_class.new(app, env) }
 
-  before do
-    allow(driver).to receive(:execute) { |*args| @cmd = args.join(' ') }
-  end
-
   after do
     sandbox.close
   end
diff --git a/test/unit/plugins/providers/docker/action/destroy_network_test.rb b/test/unit/plugins/providers/docker/action/destroy_network_test.rb
index 7b9b266da..68e8aa1bc 100644
--- a/test/unit/plugins/providers/docker/action/destroy_network_test.rb
+++ b/test/unit/plugins/providers/docker/action/destroy_network_test.rb
@@ -22,6 +22,7 @@ describe VagrantPlugins::DockerProvider::Action::DestroyNetwork do
 
   let(:machine) do
     iso_env.machine(iso_env.machine_names[0], :docker).tap do |m|
+      allow(m).to receive(:vagrantfile).and_return(vagrantfile)
       allow(m).to receive(:config).and_return(machine_config)
       allow(m.provider).to receive(:driver).and_return(driver)
       allow(m.config.vm).to receive(:networks).and_return(networks)
@@ -54,10 +55,6 @@ describe VagrantPlugins::DockerProvider::Action::DestroyNetwork do
 
   subject { described_class.new(app, env) }
 
-  before do
-    allow(driver).to receive(:execute) { |*args| @cmd = args.join(' ') }
-  end
-
   after do
     sandbox.close
   end
diff --git a/test/unit/plugins/providers/docker/action/login_test.rb b/test/unit/plugins/providers/docker/action/login_test.rb
index c1ae5a4ec..07c0745a1 100644
--- a/test/unit/plugins/providers/docker/action/login_test.rb
+++ b/test/unit/plugins/providers/docker/action/login_test.rb
@@ -28,6 +28,7 @@ describe VagrantPlugins::DockerProvider::Action::Login do
       allow(m).to receive(:id).and_return("12345")
       allow(m).to receive(:config).and_return(machine_config)
       allow(m).to receive(:provider_config).and_return(provider_config)
+      allow(m).to receive(:vagrantfile).and_return(vagrantfile)
       allow(m.provider).to receive(:driver).and_return(driver)
       allow(m.provider).to receive(:host_vm?).and_return(false)
     end
@@ -42,10 +43,6 @@ describe VagrantPlugins::DockerProvider::Action::Login do
 
   subject { described_class.new(app, env) }
 
-  before do
-    allow(driver).to receive(:execute) { |*args| @cmd = args.join(' ') }
-  end
-
   after do
     sandbox.close
   end
diff --git a/test/unit/plugins/providers/docker/action/prepare_networks_test.rb b/test/unit/plugins/providers/docker/action/prepare_networks_test.rb
index c98a40f6b..67ecc5470 100644
--- a/test/unit/plugins/providers/docker/action/prepare_networks_test.rb
+++ b/test/unit/plugins/providers/docker/action/prepare_networks_test.rb
@@ -22,6 +22,7 @@ describe VagrantPlugins::DockerProvider::Action::PrepareNetworks do
 
   let(:machine) do
     iso_env.machine(iso_env.machine_names[0], :docker).tap do |m|
+      allow(m).to receive(:vagrantfile).and_return(vagrantfile)
       allow(m).to receive(:config).and_return(machine_config)
       allow(m.provider).to receive(:driver).and_return(driver)
       allow(m.config.vm).to receive(:networks).and_return(networks)