Fix component loading bug