diff --git a/site-modules/role/manifests/loadbalancer.pp b/site-modules/role/manifests/loadbalancer.pp
index a8f658c..eced20a 100644
--- a/site-modules/role/manifests/loadbalancer.pp
+++ b/site-modules/role/manifests/loadbalancer.pp
@@ -1,12 +1,12 @@
 class role::loadbalancer (
   Integer   $ports1 = '80',
-  Integer   $ports2 = undef,
-  String    $rule1 = 'Http',
-  String    $rule2 = undef,
+  Optional[Integer]   $ports2 = undef,
+  String    $rule1 = 'http',
+  Optional[String]    $rule2 = undef,
   String    $backendserver_name1 = '',
   String    $backendserver_name2 = '',
-  Integer   $backendserver_ipaddress1 = undef,
-  Integer   $backendserver_ipaddress2 = undef,
+  Optional[String]   $backendserver_ipaddress1 = undef,
+  Optional[String]   $backendserver_ipaddress2 = undef,
   ) {
   include ::haproxy
   haproxy::listen { $rule1 :