Update pikos/tests/test_trace_function_manager.py