diff --git a/resources/rabbitmq_service/actions/run.pp b/resources/rabbitmq_service/actions/run.pp
index 02ed9cf..2ea2205 100644
--- a/resources/rabbitmq_service/actions/run.pp
+++ b/resources/rabbitmq_service/actions/run.pp
@@ -1,7 +1,7 @@
 $resource = hiera($::resource_name)
 
-$port = "${resource['input']['port']}"
-$management_port = "${resource['input']['management_port']}"
+$port = "${resource['input']['port']['value']}"
+$management_port = "${resource['input']['management_port']['value']}"
 
 class { '::rabbitmq':
   service_manage    => true,