Rename import in tests too