Corrected big bug on edit category