1 files changed, 2 insertions, 5 deletions
diff --git a/ui/src/tracks/cpu_profile/controller.ts b/ui/src/tracks/cpu_profile/controller.ts
index 070cedba9..cb850ee0a 100644
--- a/ui/src/tracks/cpu_profile/controller.ts
+++ b/ui/src/tracks/cpu_profile/controller.ts
@@ -12,7 +12,6 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import {slowlyCountRows} from '../../common/query_iterator';
 import {
   TrackController,
   trackControllerRegistry
@@ -28,13 +27,13 @@ class CpuProfileTrackController extends TrackController<Config, Data> {
   static readonly kind = CPU_PROFILE_TRACK_KIND;
   async onBoundsChange(start: number, end: number, resolution: number):
       Promise<Data> {
-    const query = `select id, ts, callsite_id from cpu_profile_stack_sample
+    const query = `select id, ts from cpu_profile_stack_sample
         where utid = ${this.config.utid}
         order by ts`;
 
     const result = await this.query(query);
 
-    const numRows = slowlyCountRows(result);
+    const numRows = +result.numRecords;
     const data: Data = {
       start,
       end,
@@ -42,13 +41,11 @@ class CpuProfileTrackController extends TrackController<Config, Data> {
       length: numRows,
       ids: new Float64Array(numRows),
       tsStarts: new Float64Array(numRows),
-      callsiteId: new Uint32Array(numRows),
     };
 
     for (let row = 0; row < numRows; row++) {
       data.ids[row] = +result.columns[0].longValues![row];
       data.tsStarts[row] = +result.columns[1].longValues![row];
-      data.callsiteId[row] = +result.columns[2].longValues![row];
     }
 
     return data;