Fix JSD2 profiling