Fix regression in test suite