diff --git a/Gemfile b/Gemfile
index 8fa4c92..a70a9da 100644
--- a/Gemfile
+++ b/Gemfile
@@ -4,7 +4,7 @@ source "https://rubygems.org"
 
 gem 'test-kitchen', '>= 3.2.2'
 gem 'kitchen-salt', '>= 0.7.2'
-gem 'kitchen-docker', :git => 'https://github.com/garethgreenaway/kitchen-docker.git', :branch => 'add_rockylinux_platform_section'
+gem 'kitchen-docker', :git => 'https://github.com/test-kitchen/kitchen-docker.git', :branch => 'main'
 
 group :vagrant do
   gem 'kitchen-vagrant'
diff --git a/kitchen.yml b/kitchen.yml
index 672f86a..1603749 100644
--- a/kitchen.yml
+++ b/kitchen.yml
@@ -116,6 +116,7 @@ platforms:
         - echo "PubkeyAcceptedAlgorithms +ssh-rsa" | tee -a /etc/ssh/sshd_config
   - name: rockylinux-9
     driver:
+      platform: centosstream
       run_command: /usr/lib/systemd/systemd
       provision_command:
         - dnf -y install crypto-policies-scripts procps-ng