diff --git a/src/undo.c b/src/undo.c
index fcc9a3b..5db25f0 100644
--- a/src/undo.c
+++ b/src/undo.c
@@ -1455,7 +1455,7 @@
 # endif
        )
 	mch_setperm(file_name, (perm & 0707) | ((perm & 07) << 3));
-# ifdef HAVE_SELINUX
+# if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
     if (buf->b_ffname != NULL)
 	mch_copy_sec(buf->b_ffname, file_name);
 # endif