Forgot to move the import to module level