diff --git a/includes/class-gf-cli-form-notification.php b/includes/class-gf-cli-form-notification.php
index d630e7f..e368351 100644
--- a/includes/class-gf-cli-form-notification.php
+++ b/includes/class-gf-cli-form-notification.php
@@ -422,7 +422,7 @@ function update( $args, $assoc_args ) {
 
 		$found = false;
 		foreach ( $notifications as $key => $notification ) {
-			if ( $notification['id'] == $notification_id ) {
+      if ( $notification['id'] == $notification_id || $notification['id'] == $new_notification['id'] ) {
 				$notifications[ $key ] = $new_notification;
 				$found = true;
 				break;
@@ -489,7 +489,7 @@ public function edit( $args, $assoc_args ) {
 
 	/**
 	 * Launches the editor, setting the content and title
-	 * 
+	 *
 	 * @since 1.0-beta-1
 	 * @access protected
 	 *