Category tree fix