diff --git a/controllers/admin/AdminAttributesGroupsController.php b/controllers/admin/AdminAttributesGroupsController.php
index 15d96fbc4..b0c03d3d3 100644
--- a/controllers/admin/AdminAttributesGroupsController.php
+++ b/controllers/admin/AdminAttributesGroupsController.php
@@ -519,7 +519,7 @@ class AdminAttributesGroupsControllerCore extends AdminController
 	 */
 	public function processSave($token)
 	{
-		if ($this->display == 'add')
+		if ($this->display == 'add' || $this->display == 'edit')
 			$this->identifier = 'id_attribute_group';
 		else
 			$this->identifier = 'id_attribute';