Fixed the imports in the code templates