diff --git a/webform_classes.module b/webform_classes.module
index 047c890..c87b706 100644
--- a/webform_classes.module
+++ b/webform_classes.module
@@ -72,7 +72,11 @@ function webform_classes_form_webform_component_edit_form_alter(&$form, $compone
 function webform_classes_webform_component_update($component) {
   $nid = $component['nid'];
   $cid = $component['cid'];
-  $class = $component['classes']['classes'];
+	if(isset($component['classes']['classes'])) {
+  	$class = $component['classes']['classes'];
+	} else {
+		$class = '';
+	}
   $result = db_update('webform_component')
     ->fields(array(
     'class'=> $class,