diff --git a/csrc/common/lib/device.cpp b/csrc/common/lib/device.cpp
index 2dcefed8..a703f40b 100644
--- a/csrc/common/lib/device.cpp
+++ b/csrc/common/lib/device.cpp
@@ -1,7 +1,7 @@
 //
 // Created by morgan on 27/09/24.
 //
-
+#include <nvml.h>
 #include "device.hpp"
 
 std::optional<nvmlDevice_t> huggingface::tgi::GetDeviceByIndex(device_index_t device)