diff --git a/backend/modules/porthijack/firewall.py b/backend/modules/porthijack/firewall.py
index 9f253bc..bafa096 100644
--- a/backend/modules/porthijack/firewall.py
+++ b/backend/modules/porthijack/firewall.py
@@ -1,7 +1,6 @@
-from ast import Delete
 import asyncio
 from typing import Dict
-from modules.porthijack.nftables import FiregexTables, FiregexFilter
+from modules.porthijack.nftables import FiregexTables
 from modules.porthijack.models import Service
 from utils.sqlite import SQLite
 
diff --git a/backend/modules/porthijack/nftables.py b/backend/modules/porthijack/nftables.py
index 13a2814..17f0a5a 100644
--- a/backend/modules/porthijack/nftables.py
+++ b/backend/modules/porthijack/nftables.py
@@ -91,8 +91,8 @@ class FiregexTables(NFTableManager):
                 target=filter["chain"],
                 id=int(filter["handle"]),
                 proto=filter["expr"][1]["match"]["left"]["payload"]["protocol"],
-                public_port=filter["expr"][1]["match"]["right"] if filter["target"] == self.prerouting_porthijack else filter["expr"][2]["mangle"]["value"],
-                proxy_port=filter["expr"][1]["match"]["right"] if filter["target"] == self.postrouting_porthijack else filter["expr"][2]["mangle"]["value"], 
+                public_port=filter["expr"][1]["match"]["right"] if filter["chain"] == self.prerouting_porthijack else filter["expr"][2]["mangle"]["value"],
+                proxy_port=filter["expr"][1]["match"]["right"] if filter["chain"] == self.postrouting_porthijack else filter["expr"][2]["mangle"]["value"], 
                 ip_int=ip_int
             ))
         return res
diff --git a/backend/routers/porthijack.py b/backend/routers/porthijack.py
index b5c7e19..1394e0b 100644
--- a/backend/routers/porthijack.py
+++ b/backend/routers/porthijack.py
@@ -41,8 +41,8 @@ db = SQLite('db/port-hijacking.db', {
     'services': {
         'service_id': 'VARCHAR(100) PRIMARY KEY',
         'active' : 'BOOLEAN NOT NULL CHECK (active IN (0, 1))',
-        'public_port': 'INT NOT NULL CHECK(port > 0 and port < 65536)',
-        'proxy_port': 'INT NOT NULL CHECK(port > 0 and port < 65536)',
+        'public_port': 'INT NOT NULL CHECK(public_port > 0 and public_port < 65536)',
+        'proxy_port': 'INT NOT NULL CHECK(proxy_port > 0 and proxy_port < 65536)',
         'name': 'VARCHAR(100) NOT NULL UNIQUE',
         'proto': 'VARCHAR(3) NOT NULL CHECK (proto IN ("tcp", "udp"))',
         'ip_int': 'VARCHAR(100) NOT NULL',
diff --git a/tests/px_test.py b/tests/px_test.py
index 22ae91f..e90e34d 100755
--- a/tests/px_test.py
+++ b/tests/px_test.py
@@ -82,6 +82,7 @@ def checkRegex(regex, should_work=True, upper=False):
                 if not server.sendCheckData(secrets.token_bytes(200) + s +  secrets.token_bytes(200)):
                     puts(f"The malicious request was successfully blocked ✔", color=colors.green)
                     n_blocked += 1
+                    time.sleep(0.5)
                     if firegex.px_get_regex(r["id"])["n_packets"] == n_blocked:
                         puts(f"The packed was reported as blocked ✔", color=colors.green)
                     else:
@@ -245,4 +246,4 @@ new_internal_port = firegex.px_get_service(service_id)["internal_port"]
 if (internal_port != new_internal_port): puts(f"Sucessfully got regenerated port {new_internal_port} ✔", color=colors.green)
 else: puts(f"Test Failed: Coundn't get internal port, or it was the same as previous ✗", color=colors.red); exit_test(1)
 
-exit_test(0)
\ No newline at end of file
+exit_test(0)