diff --git a/cmd/nvidia-ctk-installer/container/toolkit/toolkit_test.go b/cmd/nvidia-ctk-installer/container/toolkit/toolkit_test.go
index 4aad36c5..ec9dc5bd 100644
--- a/cmd/nvidia-ctk-installer/container/toolkit/toolkit_test.go
+++ b/cmd/nvidia-ctk-installer/container/toolkit/toolkit_test.go
@@ -71,45 +71,45 @@ func TestInstall(t *testing.T) {
 cdiVersion: 0.5.0
 kind: example.com/class
 devices:
-- name: all
-  containerEdits:
-    deviceNodes:
-    - path: /dev/nvidia0
-      hostPath: /host/driver/root/dev/nvidia0
-    - path: /dev/nvidiactl
-      hostPath: /host/driver/root/dev/nvidiactl
-    - path: /dev/nvidia-caps-imex-channels/channel0
-      hostPath: /host/driver/root/dev/nvidia-caps-imex-channels/channel0
-    - path: /dev/nvidia-caps-imex-channels/channel1
-      hostPath: /host/driver/root/dev/nvidia-caps-imex-channels/channel1
-    - path: /dev/nvidia-caps-imex-channels/channel2047
-      hostPath: /host/driver/root/dev/nvidia-caps-imex-channels/channel2047
+    - name: all
+      containerEdits:
+        deviceNodes:
+            - path: /dev/nvidia0
+              hostPath: /host/driver/root/dev/nvidia0
+            - path: /dev/nvidiactl
+              hostPath: /host/driver/root/dev/nvidiactl
+            - path: /dev/nvidia-caps-imex-channels/channel0
+              hostPath: /host/driver/root/dev/nvidia-caps-imex-channels/channel0
+            - path: /dev/nvidia-caps-imex-channels/channel1
+              hostPath: /host/driver/root/dev/nvidia-caps-imex-channels/channel1
+            - path: /dev/nvidia-caps-imex-channels/channel2047
+              hostPath: /host/driver/root/dev/nvidia-caps-imex-channels/channel2047
 containerEdits:
-  env:
-  - NVIDIA_VISIBLE_DEVICES=void
-  hooks:
-  - hookName: createContainer
-    path: {{ .toolkitRoot }}/nvidia-cdi-hook
-    args:
-    - nvidia-cdi-hook
-    - create-symlinks
-    - --link
-    - libcuda.so.1::/lib/x86_64-linux-gnu/libcuda.so
-  - hookName: createContainer
-    path: {{ .toolkitRoot }}/nvidia-cdi-hook
-    args:
-    - nvidia-cdi-hook
-    - update-ldcache
-    - --folder
-    - /lib/x86_64-linux-gnu
-  mounts:
-  - hostPath: /host/driver/root/lib/x86_64-linux-gnu/libcuda.so.999.88.77
-    containerPath: /lib/x86_64-linux-gnu/libcuda.so.999.88.77
-    options:
-    - ro
-    - nosuid
-    - nodev
-    - bind
+    env:
+        - NVIDIA_VISIBLE_DEVICES=void
+    hooks:
+        - hookName: createContainer
+          path: {{ .toolkitRoot }}/nvidia-cdi-hook
+          args:
+            - nvidia-cdi-hook
+            - create-symlinks
+            - --link
+            - libcuda.so.1::/lib/x86_64-linux-gnu/libcuda.so
+        - hookName: createContainer
+          path: {{ .toolkitRoot }}/nvidia-cdi-hook
+          args:
+            - nvidia-cdi-hook
+            - update-ldcache
+            - --folder
+            - /lib/x86_64-linux-gnu
+    mounts:
+        - hostPath: /host/driver/root/lib/x86_64-linux-gnu/libcuda.so.999.88.77
+          containerPath: /lib/x86_64-linux-gnu/libcuda.so.999.88.77
+          options:
+            - ro
+            - nosuid
+            - nodev
+            - bind
 `,
 		},
 	}
diff --git a/cmd/nvidia-ctk/cdi/generate/generate_test.go b/cmd/nvidia-ctk/cdi/generate/generate_test.go
index c74cff23..9e9a6a4e 100644
--- a/cmd/nvidia-ctk/cdi/generate/generate_test.go
+++ b/cmd/nvidia-ctk/cdi/generate/generate_test.go
@@ -67,51 +67,51 @@ func TestGenerateSpec(t *testing.T) {
 cdiVersion: 0.5.0
 kind: example.com/device
 devices:
-- name: "0"
-  containerEdits:
-    deviceNodes:
-    - path: /dev/nvidia0
-      hostPath: {{ .driverRoot }}/dev/nvidia0
-- name: all
-  containerEdits:
-    deviceNodes:
-    - path: /dev/nvidia0
-      hostPath: {{ .driverRoot }}/dev/nvidia0
+    - name: "0"
+      containerEdits:
+        deviceNodes:
+            - path: /dev/nvidia0
+              hostPath: {{ .driverRoot }}/dev/nvidia0
+    - name: all
+      containerEdits:
+        deviceNodes:
+            - path: /dev/nvidia0
+              hostPath: {{ .driverRoot }}/dev/nvidia0
 containerEdits:
-  env:
-  - NVIDIA_VISIBLE_DEVICES=void
-  deviceNodes:
-  - path: /dev/nvidiactl
-    hostPath: {{ .driverRoot }}/dev/nvidiactl
-  hooks:
-  - hookName: createContainer
-    path: /usr/bin/nvidia-cdi-hook
-    args:
-    - nvidia-cdi-hook
-    - create-symlinks
-    - --link
-    - libcuda.so.1::/lib/x86_64-linux-gnu/libcuda.so
-  - hookName: createContainer
-    path: /usr/bin/nvidia-cdi-hook
-    args:
-    - nvidia-cdi-hook
-    - enable-cuda-compat
-    - --host-driver-version=999.88.77
-  - hookName: createContainer
-    path: /usr/bin/nvidia-cdi-hook
-    args:
-    - nvidia-cdi-hook
-    - update-ldcache
-    - --folder
-    - /lib/x86_64-linux-gnu
-  mounts:
-  - hostPath: {{ .driverRoot }}/lib/x86_64-linux-gnu/libcuda.so.999.88.77
-    containerPath: /lib/x86_64-linux-gnu/libcuda.so.999.88.77
-    options:
-    - ro
-    - nosuid
-    - nodev
-    - bind
+    env:
+        - NVIDIA_VISIBLE_DEVICES=void
+    deviceNodes:
+        - path: /dev/nvidiactl
+          hostPath: {{ .driverRoot }}/dev/nvidiactl
+    hooks:
+        - hookName: createContainer
+          path: /usr/bin/nvidia-cdi-hook
+          args:
+            - nvidia-cdi-hook
+            - create-symlinks
+            - --link
+            - libcuda.so.1::/lib/x86_64-linux-gnu/libcuda.so
+        - hookName: createContainer
+          path: /usr/bin/nvidia-cdi-hook
+          args:
+            - nvidia-cdi-hook
+            - enable-cuda-compat
+            - --host-driver-version=999.88.77
+        - hookName: createContainer
+          path: /usr/bin/nvidia-cdi-hook
+          args:
+            - nvidia-cdi-hook
+            - update-ldcache
+            - --folder
+            - /lib/x86_64-linux-gnu
+    mounts:
+        - hostPath: {{ .driverRoot }}/lib/x86_64-linux-gnu/libcuda.so.999.88.77
+          containerPath: /lib/x86_64-linux-gnu/libcuda.so.999.88.77
+          options:
+            - ro
+            - nosuid
+            - nodev
+            - bind
 `,
 		},
 	}
diff --git a/pkg/nvcdi/lib-imex_test.go b/pkg/nvcdi/lib-imex_test.go
index e928fddd..fb7fc1eb 100644
--- a/pkg/nvcdi/lib-imex_test.go
+++ b/pkg/nvcdi/lib-imex_test.go
@@ -41,24 +41,24 @@ func TestImexMode(t *testing.T) {
 cdiVersion: 0.5.0
 kind: nvidia.com/imex-channel
 devices:
-- name: "0"
-  containerEdits:
-    deviceNodes:
-    - path: /dev/nvidia-caps-imex-channels/channel0
-      hostPath: {{ .hostRoot }}/dev/nvidia-caps-imex-channels/channel0
-- name: "1"
-  containerEdits:
-    deviceNodes:
-    - path: /dev/nvidia-caps-imex-channels/channel1
-      hostPath: {{ .hostRoot }}/dev/nvidia-caps-imex-channels/channel1
-- name: "2047"
-  containerEdits:
-    deviceNodes:
-    - path: /dev/nvidia-caps-imex-channels/channel2047
-      hostPath: {{ .hostRoot }}/dev/nvidia-caps-imex-channels/channel2047
+    - name: "0"
+      containerEdits:
+        deviceNodes:
+            - path: /dev/nvidia-caps-imex-channels/channel0
+              hostPath: {{ .hostRoot }}/dev/nvidia-caps-imex-channels/channel0
+    - name: "1"
+      containerEdits:
+        deviceNodes:
+            - path: /dev/nvidia-caps-imex-channels/channel1
+              hostPath: {{ .hostRoot }}/dev/nvidia-caps-imex-channels/channel1
+    - name: "2047"
+      containerEdits:
+        deviceNodes:
+            - path: /dev/nvidia-caps-imex-channels/channel2047
+              hostPath: {{ .hostRoot }}/dev/nvidia-caps-imex-channels/channel2047
 containerEdits:
-  env:
-  - NVIDIA_VISIBLE_DEVICES=void
+    env:
+        - NVIDIA_VISIBLE_DEVICES=void
 `
 	expectedSpec = strings.ReplaceAll(expectedSpec, "{{ .hostRoot }}", hostRoot)
 
diff --git a/pkg/nvcdi/spec/spec_test.go b/pkg/nvcdi/spec/spec_test.go
index f5d50608..f95cfe3e 100644
--- a/pkg/nvcdi/spec/spec_test.go
+++ b/pkg/nvcdi/spec/spec_test.go
@@ -54,10 +54,10 @@ func TestSpec(t *testing.T) {
 cdiVersion: 0.8.0
 kind: nvidia.com/gpu
 devices:
-- name: one
-  containerEdits:
-    env:
-    - DEVICE_FOO=bar
+    - name: one
+      containerEdits:
+        env:
+            - DEVICE_FOO=bar
 `,
 		},
 		{
@@ -80,10 +80,10 @@ devices:
 cdiVersion: 0.5.0
 kind: nvidia.com/gpu
 devices:
-- name: one
-  containerEdits:
-    env:
-    - DEVICE_FOO=bar
+    - name: one
+      containerEdits:
+        env:
+            - DEVICE_FOO=bar
 `,
 		},
 		{
@@ -105,10 +105,10 @@ devices:
 cdiVersion: 0.3.0
 kind: nvidia.com/gpu
 devices:
-- name: one
-  containerEdits:
-    env:
-    - DEVICE_FOO=bar
+    - name: one
+      containerEdits:
+        env:
+            - DEVICE_FOO=bar
 `,
 		},
 		{
@@ -138,14 +138,14 @@ devices:
 cdiVersion: 0.5.0
 kind: nvidia.com/gpu
 devices:
-- name: one
-  containerEdits:
-    env:
-    - DEVICE_FOO=bar
+    - name: one
+      containerEdits:
+        env:
+            - DEVICE_FOO=bar
 containerEdits:
-  deviceNodes:
-  - path: /dev/dev0
-    hostPath: /some/dev/dev0
+    deviceNodes:
+        - path: /dev/dev0
+          hostPath: /some/dev/dev0
 `,
 		},
 		{
@@ -183,14 +183,14 @@ containerEdits:
 cdiVersion: 0.5.0
 kind: nvidia.com/gpu
 devices:
-- name: one
-  containerEdits:
-    env:
-    - DEVICE_FOO=bar
+    - name: one
+      containerEdits:
+        env:
+            - DEVICE_FOO=bar
 containerEdits:
-  deviceNodes:
-  - path: /dev/dev0
-    hostPath: /dev/dev0
+    deviceNodes:
+        - path: /dev/dev0
+          hostPath: /dev/dev0
 `,
 		},
 	}