Restore perf functionality in command unit test