Test runner fix