Fix profiling unit tests