Updated import to be relative to the package