diff --git a/manifests/init.pp b/manifests/init.pp
index 83c2d74..ca30ee1 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -115,5 +115,4 @@
       subscribe   => File['colorprompt.sh'],
     }
   }
-
 }
diff --git a/manifests/params.pp b/manifests/params.pp
index e69eb9c..e606252 100644
--- a/manifests/params.pp
+++ b/manifests/params.pp
@@ -1,6 +1,5 @@
 # Default parameters
 class colorprompt::params {
-
   $ensure            = 'present'
   $path              = '/etc/profile.d/colorprompt.sh'
   $default_usercolor = 'cyan'
@@ -9,8 +8,7 @@
   $env_name          = undef
   $env_color         = undef
 
-  case $::osfamily {
-
+  case $::facts['os']['family'] {
     'RedHat': {
       $prompt      = '${env}[${userColor}\u\[\e[0m\]@${serverColor}\h\[\e[0m\] \W]\\\\\\$ '
       $modify_skel = false
@@ -24,9 +22,7 @@
     }
 
     default: {
-      fail("Unsupported osfamily: ${::osfamily}")
+      fail("Unsupported os family: ${::facts['os']['family']}")
     }
-
   }
-
 }