diff --git a/tests/bench_raytracer.nim b/tests/bench_raytracer.nim
index d25f6b5..5ef9d41 100644
--- a/tests/bench_raytracer.nim
+++ b/tests/bench_raytracer.nim
@@ -2,7 +2,7 @@
 ## MIT License
 ## Copyright (c) 2021 Edin Omeragic
 
-import benchy, chroma, math, pixie, times, vmath
+import benchy, chroma, math, pixie, vmath
 
 {.push inline, noinit, checks: off.}
 
@@ -251,7 +251,7 @@ proc renderScene(scene: Scene, sceneImage: Image) =
     var pos = y * w
     for x in 0 ..< w:
       ray.dir = getPoint(x, y, scene.camera, h, w)
-      sceneImage.setRgbaUnsafe(x, y, scene.traceRay(ray, 0))
+      sceneImage.unsafe[x, y] = scene.traceRay(ray, 0).asRgbx()
       pos = pos + 1
 
 proc render(): Image =
diff --git a/tests/bench_raytracer_glm.nim b/tests/bench_raytracer_glm.nim
index 04cbf4d..646d586 100644
--- a/tests/bench_raytracer_glm.nim
+++ b/tests/bench_raytracer_glm.nim
@@ -2,8 +2,8 @@
 ## MIT License
 ## Copyright (c) 2021 Edin Omeragic
 
-import benchy, chroma, math, times, glm
-from pixie import Image, newImage, writeFile, setRgbaUnsafe
+import benchy, chroma, math, glm
+from pixie import Image, newImage, writeFile, dataIndex
 
 type Vec3 = glm.Vec3[float32]
 
@@ -254,7 +254,7 @@ proc renderScene(scene: Scene, sceneImage: Image) =
     var pos = y * w
     for x in 0 ..< w:
       ray.dir = getPoint(x, y, scene.camera, h, w)
-      sceneImage.setRgbaUnsafe(x, y, scene.traceRay(ray, 0))
+      sceneImage.data[sceneImage.dataIndex(x, y)] = scene.traceRay(ray, 0).asRgbx()
       pos = pos + 1
 
 proc render(): Image =