Fix profiling for MapStep