diff --git a/firewall/firewall-add b/firewall/firewall-add
index 1b18792..f474867 100755
--- a/firewall/firewall-add
+++ b/firewall/firewall-add
@@ -52,7 +52,7 @@ fi;
 
 
 prerouting() {
-	LINES=$($IPTABLES -L --line-number -n | grep PREROUTING | grep $SOURCE_PORT |grep $TARGET_IP |grep $TARGET_PORT |grep $COMMENT | awk '{print $1}'| tac)
+	LINES=$($IPTABLES -L --line-number -n | grep DNAT | grep $SOURCE_PORT |grep $TARGET_IP |grep $TARGET_PORT |grep $COMMENT | awk '{print $1}'| tac)
 	debug "Previous prerouting lines: "$LINES
 	# DELETE UNECESSARY LINES FROM PREVIOUS RULES
 	if [ -n "$LINES" ] ; then
@@ -81,7 +81,7 @@ postrouting() {
 	fi
 	
 	if  [ -n "$SOURCE_IP" ] ; then
-		LINES=$($IPTABLES -L --line-number -n | grep POSTROUTING | grep $COMMENT | grep $SOURCE_IP_FOR_POSTROUTING | grep $SOURCE_PORT | awk '{print $1}'| tac)
+		LINES=$($IPTABLES -L --line-number -n | grep MASQUERADE | grep $COMMENT | grep $SOURCE_IP_FOR_POSTROUTING | grep $SOURCE_PORT | awk '{print $1}'| tac)
 		debug "Previous postrouting lines: "$LINES
 
 		# DELETE UNECESSARY LINES FROM PREVIOUS RULES