diff --git a/plugins/providers/docker/action/wait_for_running.rb b/plugins/providers/docker/action/wait_for_running.rb
index 89164ddcb..479af8a14 100644
--- a/plugins/providers/docker/action/wait_for_running.rb
+++ b/plugins/providers/docker/action/wait_for_running.rb
@@ -16,8 +16,10 @@ module VagrantPlugins
 
           wait = true
           if !machine.provider_config.remains_running
+            @logger.debug("remains_running is false")
             wait = false
-          elsif machine.provider.state.id == :running
+          elsif machine.state.id == :running
+            @logger.debug("container is already running")
             wait = false
           end
 
@@ -28,7 +30,7 @@ module VagrantPlugins
 
           # First, make sure it leaves the stopped state if its supposed to.
           after = sleeper(5)
-          while machine.provider.state.id == :stopped
+          while machine.state.id == :stopped
             if after[:done]
               raise Errors::StateStopped
             end
@@ -38,7 +40,7 @@ module VagrantPlugins
           # Then, wait for it to become running
           after = sleeper(30)
           while true
-            state = machine.provider.state
+            state = machine.state
             break if state.id == :running
             @logger.info("Waiting for container to run. State: #{state.id}")