Fix missing includes from test_runner