Fix an import in tests