Fix another old import