diff --git a/install/install.php b/install/install.php
index e657f69509b2d264a1c14c0b8da94d75f448f238..f84c5d8f39e63edba09030a69a80705adea7490c 100644
--- a/install/install.php
+++ b/install/install.php
@@ -229,19 +229,17 @@ if($install_mode == 'standard') {
 	$inst->configure_apps_vhost();
     
 	//* Configure Firewall
-	//swriteln('Configuring Firewall');
-	//$inst->configure_firewall();
 	//** Configure Firewall	
-	if($conf['bastille']['installed'] == true) {
+	if($conf['ufw']['installed'] == true) {
+		//* Configure UFW Firewall
+		$conf['services']['firewall'] = true;
+		swriteln('Configuring UFW Firewall');
+		$inst->configure_ufw_firewall();
+	} else {
 		//* Configure Bastille Firewall
 		$conf['services']['firewall'] = true;
 		swriteln('Configuring Bastille Firewall');
 		$inst->configure_firewall();
-	} elseif($conf['ufw']['installed'] == true) {
-		//* Configure Ubuntu Firewall
-		$conf['services']['firewall'] = true;
-		swriteln('Configuring Ubuntu Firewall');
-		$inst->configure_ufw_firewall();
 	}
 
     //* Configure Fail2ban