diff --git a/salt/firewall/init.sls b/salt/firewall/init.sls
index 34e69739f..18d0c814d 100644
--- a/salt/firewall/init.sls
+++ b/salt/firewall/init.sls
@@ -1,19 +1,11 @@
 # Firewall Magic for the grid
-
-{% if grains['role'] == 'so-master' or grains['role'] == 'so-eval' %}
-
+{%- if grains['role'] == 'so-master' or grains['role'] == 'so-eval' %}
 {%- set ip = salt['pillar.get']('static:masterip', '') %}
-
-{% elif grains['role'] == 'so-node'%}
-
+{%- elif grains['role'] == 'so-node' %}
 {%- set ip = salt['pillar.get']('node:mainip', '') %}
-
-{% elif grains['role'] == 'so-sensor'%}
-
-{%- set ip = salt['pillar.get']('node:mainip', '') %}
-
-{% endif %}
-
+{%- elif grains['role'] == 'so-sensor' %}
+{%- set ip = salt['pillar.get']('sensor:mainip', '') %}
+{%- endif %}
 # Keep localhost in the game
 iptables_allow_localhost:
   iptables.append:
diff --git a/so-setup-network.sh b/so-setup-network.sh
index 2a8d3c127..ac4694f66 100644
--- a/so-setup-network.sh
+++ b/so-setup-network.sh
@@ -453,7 +453,11 @@ get_filesystem_nsm(){
 
 get_log_size_limit() {
 
-  DISK_SIZE_K=`df /nsm |grep -v "^Filesystem" | awk '{print $2}'`
+  DISK_DIR="/"
+  if [ -d /nsm ]; then
+    DISK_DIR="/nsm"
+  fi
+  DISK_SIZE_K=`df $DISK_DIR |grep -v "^Filesystem" | awk '{print $2}'`
   PERCENTAGE=85
   DISK_SIZE=DISK_SIZE_K*1000
   PERCENTAGE_DISK_SPACE=`echo $(($DISK_SIZE*$PERCENTAGE/100))`