diff --git a/manifests/swift/proxy.pp b/manifests/swift/proxy.pp
index 4eaea7b..d461493 100644
--- a/manifests/swift/proxy.pp
+++ b/manifests/swift/proxy.pp
@@ -32,10 +32,12 @@ class openstack::swift::proxy (
   } else {
     $real_keystone_host = $keystone_host
   }
-  class { 'swift':
-    swift_hash_suffix => $swift_hash_suffix,
-    package_ensure    => $package_ensure,
-  }
+
+  ensure_resource('class', 'swift',
+    { swift_hash_suffix => $swift_hash_suffix,
+      package_ensure    => $package_ensure,
+    }
+  )
 
   if $memcached {
     class { 'memcached':
diff --git a/manifests/swift/storage-node.pp b/manifests/swift/storage-node.pp
index 662db0c..6e12e33 100644
--- a/manifests/swift/storage-node.pp
+++ b/manifests/swift/storage-node.pp
@@ -12,12 +12,11 @@ class openstack::swift::storage-node (
   $byte_size            = '1024',
 ) {
 
-  if !defined(swift){
-    class { 'swift':
-      swift_hash_suffix => $swift_hash_suffix,
+  ensure_resource('class', 'swift',
+    { swift_hash_suffix => $swift_hash_suffix,
       package_ensure    => $package_ensure,
     }
-  }
+  )
 
   case $storage_type {
     'loopback': {