Fix diagram to match my actual source structure