diff --git a/tests/pillar/repo_rocky.sls b/tests/pillar/repo_rocky.sls
index fa26377..e9e160b 100644
--- a/tests/pillar/repo_rocky.sls
+++ b/tests/pillar/repo_rocky.sls
@@ -3,6 +3,6 @@
     enabled: true
     repo:
       mirantis_openstack_rocky:
-        source: "deb http://mirror.mirantis.com/nightly/openstack-rocky/xenial xenial main"
+        source: "deb http://mirror.mirantis.com/nightly/openstack-rocky/{{ grains.get('oscodename') }} {{ grains.get('oscodename') }} main"
         architectures: amd64
-        key_url: "http://mirror.mirantis.com/nightly/openstack-rocky/xenial/archive-openstack-rocky.key"
\ No newline at end of file
+        key_url: "http://mirror.mirantis.com/nightly/openstack-rocky/{{ grains.get('oscodename') }}/archive-openstack-rocky.key"
\ No newline at end of file