diff --git a/chimera/agent/src/nvml/mod.rs b/chimera/agent/src/nvml/mod.rs
index adb178a6..55ace647 100644
--- a/chimera/agent/src/nvml/mod.rs
+++ b/chimera/agent/src/nvml/mod.rs
@@ -1,4 +1,4 @@
-use nvml_wrapper::Nvml;
+use nvml_wrapper::{Nvml, enum_wrappers::device::TemperatureSensor};
 use std::error::Error;
 
 #[derive(Default, Clone)]
@@ -13,7 +13,7 @@ pub fn collect_metrics() -> Result<TelemetryData, Box<dyn Error>> {
     let nvml = Nvml::init()?;
     let device = nvml.device_by_index(0)?;
     let memory = device.memory_info()?;
-    let temperature = device.temperature(0)?;
+    let temperature = device.temperature(TemperatureSensor::Gpu)?;
     let ecc_errors = 0u64; // placeholder until NVML bindings available
     Ok(TelemetryData {
         gpu_index: 0,