Fixing Edit Category bug