Fix tests for CodeStatistics