Fix bug in test suite