diff --git a/lib/kitchen/docker/helpers/cli_helper.rb b/lib/kitchen/docker/helpers/cli_helper.rb
index 8e3086b..fe3acc4 100644
--- a/lib/kitchen/docker/helpers/cli_helper.rb
+++ b/lib/kitchen/docker/helpers/cli_helper.rb
@@ -107,7 +107,7 @@ def docker_sudo_opts(options = {})
         # @param cmd [String] the command to run
         # @param options [Hash] shell-out options
         # @return [String] combined stdout and stderr
-        # @raise [Kitchen::ShellCommandFailed] if the command exits non-zero
+        # @raise [Kitchen::ShellOut::ShellCommandFailed] if the command exits non-zero
         def run_command(cmd, options = {})
           if options.fetch(:use_sudo, false)
             cmd = "#{options.fetch(:sudo_command, "sudo -E")} #{cmd}"
diff --git a/lib/kitchen/transport/docker.rb b/lib/kitchen/transport/docker.rb
index fe916b2..9669ac7 100644
--- a/lib/kitchen/transport/docker.rb
+++ b/lib/kitchen/transport/docker.rb
@@ -83,6 +83,8 @@ class DockerFailed < TransportFailed; end
       # than from Test Kitchen's configuration.
       #
       # @param state [Hash] instance state naming the container
+      # @param block [Proc, nil] forwarded to the connection's constructor, which
+      #   yields the new connection to it; the connection is not closed afterwards
       # @yieldparam connection [Connection] if a block is given
       # @return [Connection]
       def connection(state, &block)