Fix broken regression test