Arrange the import statements

Reset the import statement due to the change of directory system.