added missing collection to cleanup method