Fix regression test bug :)