Test suite fix