Test suite fixes