[HierarchyBuilder] Register extension

Change-Id: Iee6e44c7fd1ac11946a4bd21bc9dc58776fc1ef3