Add test_mode parameter.