Fix regression test