Move import out of the function