Slighly change test runner