change up import logic