Fix a regression test