Fixing another test affected by package changes