Allow test runner to take args
Fix import issue with logging handler