Component diagram fix