Fix old import