Update Test framework. Minor fix.