diff --git a/block_class.module b/block_class.module
index 9dc7386..2895633 100644
--- a/block_class.module
+++ b/block_class.module
@@ -91,11 +91,11 @@ function block_class_form_alter(&$form, &$form_state, $form_id) {
       '#maxlength' => 255,
     );
 
-    $form['#submit'][] = 'block_class_form_submit';
+    //$form['#submit'][] = 'block_class_form_submit';
+    $form['actions']['submit']['#submit'][] = 'block_class_form_submit';
   }
 }
 
-
 /**
  * Save supplied class.
  */
