X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=src%2Fscript.c;h=e5b83415561729f77e72109b6986234d497ab38b;hb=1234ab04386fd63eafb19ecf245b730d6461e869;hp=3cc7e2b47c663afacb4575acf3baa8147af69c32;hpb=eb0b01e94374774609c31ab4508d8cfca8a400e1;p=odhcp6c.git

diff --git a/src/script.c b/src/script.c
index 3cc7e2b..e5b8341 100644
--- a/src/script.c
+++ b/src/script.c
@@ -187,9 +187,9 @@ static void entry_to_env(const char *name, const void *data, size_t len, enum en
 			} else {
 				buf_len += snprintf(&buf[buf_len], 24, ",%u,%u", e[i].preferred, e[i].valid);
 			}
-                        if (type == ENTRY_PREFIX && e[i].prefix_class) {
-                          buf_len += snprintf(&buf[buf_len], 12, ",class=%u", e[i].prefix_class);
-                        }
+
+			if ((type == ENTRY_PREFIX || type == ENTRY_ADDRESS) && e[i].class)
+				buf_len += snprintf(&buf[buf_len], 12, ",class=%u", e[i].class);
 
 			if (type == ENTRY_PREFIX && e[i].priority) {
 				// priority and router are abused for prefix exclusion