fix for category remove issue