diff --git a/mail_edit.admin.inc b/mail_edit.admin.inc
index 93884c7..2f96c78 100644
--- a/mail_edit.admin.inc
+++ b/mail_edit.admin.inc
@@ -461,7 +461,7 @@ function mail_edit_template_subform($template) {
     );
 
     $form['rewrite_plaintext'] = array(
-      '#type' =>'checkbox',
+      '#type' => 'checkbox',
       '#title' => t('Rewrite plaintext'),
       '#default_value' => isset($template['rewrite_plaintext']) ? $template['rewrite_plaintext'] : '',
     );
diff --git a/mail_edit.install b/mail_edit.install
index 1ca4824..147a783 100644
--- a/mail_edit.install
+++ b/mail_edit.install
@@ -159,7 +159,12 @@ function mail_edit_update_7001() {
 /**
  * Add 'plaintext' column to 'mail_edit' table.
  */
-function mail_edit_update_7101() {
-  db_add_field('mail_edit', 'rewrite_plaintext', array('type' => 'int', 'not null' => TRUE, 'default' => 0));
-  db_add_field('mail_edit', 'plaintext', array('type' => 'text', 'size' => 'normal'));
+function mail_edit_update_7002() {
+  $schema = mail_edit_schema();
+  if (!db_field_exists('mail_edit', 'rewrite_plaintext')) {
+    db_add_field('mail_edit', 'rewrite_plaintext', $schema['mail_edit']['fields']['rewrite_plaintext']);
+  }
+  if (!db_field_exists('mail_edit', 'plaintext')) {
+    db_add_field('mail_edit', 'plaintext', $schema['mail_edit']['fields']['plaintext']);
+  }
 }