Stop using a deprecated import