diff --git a/site-modules/profile/manifests/prometheus/apt_config.pp b/site-modules/profile/manifests/prometheus/apt_config.pp
index c8fe254d..9ecb8b48 100644
--- a/site-modules/profile/manifests/prometheus/apt_config.pp
+++ b/site-modules/profile/manifests/prometheus/apt_config.pp
@@ -1,22 +1,22 @@
 # APT configuration for prometheus
 class profile::prometheus::apt_config {
-  if $facts['os']['distro']['codename'] == 'stretch' {
+  if $facts['lsbdistcodename'] == 'stretch' {
     $pinned_packages = [
       'prometheus',
       'prometheus-alertmanager',
       'prometheus-node-exporter',
     ]
 
     ::apt::pin {'prometheus':
       explanation => 'Pin prometheus to backports',
       codename    => 'stretch-backports',
       packages    => $pinned_packages,
       priority    => 990,
     }
 
   } else {
     ::apt::pin {'prometheus':
       ensure => absent
     }
   }
 }