Fix test suite