Make profiling work pt 2