diff --git a/aten/src/ATen/native/vulkan/glsl/unsqueeze.glsl b/aten/src/ATen/native/vulkan/glsl/unsqueeze.glsl
index 8512ca8856a..213d01edb1a 100644
--- a/aten/src/ATen/native/vulkan/glsl/unsqueeze.glsl
+++ b/aten/src/ATen/native/vulkan/glsl/unsqueeze.glsl
@@ -41,12 +41,10 @@ void main() {
   if (dim == 1) {
     int src_x = pos.x;
     int src_y = pos.y;
-    for (int i = 0; i < 4; i++) {
-      int src_z = pos.z / (channels * 4);
-      int p = (pos.z / channels) % 4;
-      const vec4 v = texelFetch(uImage, ivec3(src_x, src_y, src_z), 0);
-      out_texel[i] = v[p];
-    }
+    int src_z = pos.z / 4;
+    int p = pos.z % 4;
+    const vec4 v = texelFetch(uImage, ivec3(src_x, src_y, src_z), 0);
+    out_texel[0] = v[p];
     imageStore(uOutput, pos, out_texel);
   } else if (dim == 2) {
     int src_x = pos.x;