Fix test runner for local testing