diff --git a/manifests/gc.pp b/manifests/gc.pp
index 58094dc..05dcbf1 100644
--- a/manifests/gc.pp
+++ b/manifests/gc.pp
@@ -24,8 +24,8 @@
   $user    = $stash::user,
   $homedir = $stash::homedir,
 ) {
-  if $::stash_version {
-    if versioncmp($::stash_version, '3.2') < 0 {
+  if $facts['stash_version'] {
+    if versioncmp($facts['stash_version'], '3.2') < 0 {
       $shared = ''
     } else {
       $shared = '/shared'
diff --git a/manifests/init.pp b/manifests/init.pp
index 27aaf8e..fd3cbfb 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -81,10 +81,10 @@
 ) inherits stash::params {
   Exec { path => ['/bin/', '/sbin/' , '/usr/bin/', '/usr/sbin/'] }
 
-  if $::stash_version {
+  if $facts['stash_version'] {
     # If the running version of stash is less than the expected version of stash
     # Shut it down in preparation for upgrade.
-    if versioncmp($version, $::stash_version) > 0 {
+    if versioncmp($version, $facts['stash_version']) > 0 {
       notify { 'Attempting to upgrade stash': }
       exec { $stop_stash: }
       if versioncmp($version, '3.2.0') > 0 {