commit 272503b: [Minor] Fix corner cases

[Vsevolod Stakhov]

Author: Vsevolod Stakhov
Date: 2022-10-22 13:17:45 +0100
URL: https://github.com/rspamd/rspamd/commit/272503b1ba88fe9802192ef65a1f7674e0a51f2b (HEAD -> master)

[Minor] Fix corner cases

---
 lualib/lua_mime.lua | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lualib/lua_mime.lua b/lualib/lua_mime.lua
index 9dff5cfed..038dccceb 100644
--- a/lualib/lua_mime.lua
+++ b/lualib/lua_mime.lua
@@ -590,8 +590,8 @@ exports.modify_headers = function(task, hdr_alterations, mode)
 
   if mode == 'compat' then
     -- Clear empty alterations in the compat mode
-    if not next(add_headers) then add_headers = nil end
-    if not next(hdr_alterations.remove) then hdr_alterations.remove = nil end
+    if add_headers and not next(add_headers) then add_headers = nil end
+    if hdr_alterations.remove and not next(hdr_alterations.remove) then hdr_alterations.remove = nil end
   end
   task:set_milter_reply({
     add_headers = add_headers,