diff --git a/nodeformcols.module b/nodeformcols.module
index 5db4eae..836da37 100644
--- a/nodeformcols.module
+++ b/nodeformcols.module
@@ -194,8 +194,8 @@ function template_preprocess_node_form(&$aVars) {
     );
   }
 
-  foreach ($has_elements as $name => $has) {
-    if ($has) {
+  foreach ($regions as $name => $data) {
+    if ($has_elements[$name]) {
       $class[] = 'node-form-has-region-' . $name;
       $form['nodeformcols_region_' . $name] = $regions[$name];
     }