1 files changed, 3 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/tsg_vgpu.c b/drivers/gpu/nvgpu/vgpu/tsg_vgpu.c
index a81b5022..3553bf51 100644
--- a/drivers/gpu/nvgpu/vgpu/tsg_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/tsg_vgpu.c
@@ -21,11 +21,12 @@
  */
 
 #include "gk20a/gk20a.h"
-#include "gk20a/tsg_gk20a.h"
 #include "fifo_vgpu.h"
-#include <nvgpu/channel.h>
 
+#include <nvgpu/channel.h>
+#include <nvgpu/tsg.h>
 #include <nvgpu/bug.h>
+
 #include <nvgpu/vgpu/tegra_vgpu.h>
 #include <nvgpu/vgpu/vgpu.h>