Tiny change to import statement