diff --git a/includes/class-gf-cli-form-notification.php b/includes/class-gf-cli-form-notification.php
index d630e7f..20d0c00 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 || $notifications['id'] == $new_notification['id'] ) {
 				$notifications[ $key ] = $new_notification;
 				$found = true;
 				break;