diff --git a/reference/opt/shaders-hlsl/comp/rayquery.fxconly.comp b/reference/opt/shaders-hlsl/comp/rayquery.fxconly.comp
index 8319dfb1..a18be3bb 100644
--- a/reference/opt/shaders-hlsl/comp/rayquery.fxconly.comp
+++ b/reference/opt/shaders-hlsl/comp/rayquery.fxconly.comp
@@ -1,5 +1,5 @@
 RWByteAddressBuffer _17 : register(u0);
-uniform RaytracingAccelerationStructure rtas;
+uniform RaytracingAccelerationStructure rtas : register(t1);
 
 static RayQuery<RAY_FLAG_NONE> rayQuery;
 
diff --git a/reference/shaders-hlsl/comp/rayquery.fxconly.comp b/reference/shaders-hlsl/comp/rayquery.fxconly.comp
index 48a99d00..1339f45f 100644
--- a/reference/shaders-hlsl/comp/rayquery.fxconly.comp
+++ b/reference/shaders-hlsl/comp/rayquery.fxconly.comp
@@ -8,7 +8,7 @@ struct Ray
 
 RWByteAddressBuffer _17 : register(u0);
 RWByteAddressBuffer _257 : register(u2);
-uniform RaytracingAccelerationStructure rtas;
+uniform RaytracingAccelerationStructure rtas : register(t1);
 
 static RayQuery<RAY_FLAG_NONE> rayQuery;