diff --git a/manifests/params.pp b/manifests/params.pp
index 33adb874a..a8d9ac11c 100644
--- a/manifests/params.pp
+++ b/manifests/params.pp
@@ -223,11 +223,12 @@
       }
 
       $python_package_name = 'python-mysqldb'
-      $ruby_package_name   = $::operatingsystemrelease ? {
-        '8'                => 'ruby-mysql',
-        '9'                => 'ruby-mysql2',
-        '14'               => 'ruby-mysql',
-        '16'               => 'ruby-mysql',
+      $ruby_package_name   = $::lsbdistcodename ? {
+        'jessie'           => 'ruby-mysql',
+        'stretch'          => 'ruby-mysql2',
+        'trusty'           => 'ruby-mysql',
+        'xenial'           => 'ruby-mysql',
+        'bionic'           => 'ruby-mysql2',
         default            => 'libmysql-ruby',
       }
     }