Fix Class diagram creation from component