1 files changed, 4 insertions, 4 deletions

diff --git a/device_info_procfs.c b/device_info_procfs.c
index 168905f..8fe9709 100644
--- a/device_info_procfs.c
+++ b/device_info_procfs.c
@@ -1,4 +1,4 @@
-#include "nvdebug.h"
+#include "nvdebug_linux.h"
 #include <linux/seq_file.h> // For seq_* functions and types
 #include <linux/uaccess.h> // For copy_to_user()
 
@@ -16,7 +16,7 @@ static ssize_t nvdebug_reg32_read(struct file *f, char __user *buf, size_t size,
         if (size < 16 || *off != 0)
                 return 0;
         // 32 bit register will always take less than 16 characters to print
-        chars_written = scnprintf(out, 16, "%#0x\n", nvdebug_readl(g, (uintptr_t)PDE_DATA(file_inode(f))));
+        chars_written = scnprintf(out, 16, "%#0x\n", nvdebug_readl(g, (uintptr_t)pde_data(file_inode(f))));
         if (copy_to_user(buf, out, chars_written))
                 printk(KERN_WARNING "Unable to copy all data for %s\n", file_dentry(f)->d_name.name);
         *off += chars_written;
@@ -30,7 +30,7 @@ static ssize_t nvdebug_reg_range_read(struct file *f, char __user *buf, size_t s
         struct nvdebug_state *g = &g_nvdebug_state[file2parentgpuidx(f)];
         // See comment in nvdebug_entry.c to understand `union reg_range`
         union reg_range range;
-        range.raw = (uintptr_t)PDE_DATA(file_inode(f));
+        range.raw = (uintptr_t)pde_data(file_inode(f));
 
         // "0x" + up to 32-bit register as hex + "\n\0" is at most 12 characters
         if (size < 12 || *off != 0)
@@ -55,7 +55,7 @@ struct file_operations nvdebug_read_reg32_file_ops = {
 };
 
 // Generic mechanism used for printing a subset of bits from a register
-// Please store a `union reg_range` rather than a `uintptr_t` in the PDE_DATA
+// Please store a `union reg_range` rather than a `uintptr_t` in the pde_data
 struct file_operations nvdebug_read_reg_range_file_ops = {
         .read = nvdebug_reg_range_read,
         .llseek = default_llseek,