Small bug fix in test suite