Merge pull request #465 from Codeinwp/bugfix/pro/549

Fixed broken product page layout

Author: vytisbulkevicius

classes/form.class.php

@@ -182,6 +182,10 @@ function( $collapse_field ) {
 
 				$section_started = true;
 				$ppom_collapse_counter ++;
+
+				if ( count( $fields ) === $ppom_field_counter ) {
+					echo '</div>';
+				}
 			}
 
 			// skip collapse div
@@ -218,14 +222,14 @@ function( $collapse_field ) {
 
 			$field_html .= '</div>';
 
-			if ( count( $collapse_fields ) === $ppom_collapse_counter && $section_started ) {
-				$section_started = false;
-			}
-
 			// Filter: nmforms_input_htmls
 			// @TODO need to change with relevant name
 			echo apply_filters( 'nmforms_input_html', $field_html, $meta, $default_value );
 
+			if ( count( $fields ) === $ppom_field_counter && $section_started ) {
+				echo '</div>';
+				$section_started = false;
+			}
 		}
 	}