diff --git a/templates/lxc-debian.in b/templates/lxc-debian.in
index c2371ea..8e3b241 100644
--- a/templates/lxc-debian.in
+++ b/templates/lxc-debian.in
@@ -240,7 +240,7 @@ write_sourceslist()
     if [ "$mainonly" = 1 ]; then
       non_main=''
     else
-        case "$release" in
+        case "$distro_codename" in
           wheezy|jessie|stretch|buster|bullseye)
             non_main=' contrib non-free'
             ;;