Fix component reference