Minor fix in test framework replacement