Update import to reflect changes in the directory structure.