Fixes regression test