diff --git a/build.zig b/build.zig
index 764db3438..c6c461b4c 100644
--- a/build.zig
+++ b/build.zig
@@ -271,14 +271,6 @@ pub fn build(b: *std.Build) !void {
         if (config.emit_test_exe) b.installArtifact(test_exe);
         _ = try deps.add(test_exe);
 
-        // Only need ziglyph for tests
-        if (b.lazyDependency("ziglyph", .{
-            .target = test_exe.root_module.resolved_target.?,
-            .optimize = test_exe.root_module.optimize.?,
-        })) |dep| {
-            test_exe.root_module.addImport("ziglyph", dep.module("ziglyph"));
-        }
-
         // Normal test running
         const test_run = b.addRunArtifact(test_exe);
         test_step.dependOn(&test_run.step);
diff --git a/src/build/SharedDeps.zig b/src/build/SharedDeps.zig
index 9461d48b7..7c0619b5e 100644
--- a/src/build/SharedDeps.zig
+++ b/src/build/SharedDeps.zig
@@ -403,6 +403,14 @@ pub fn add(
     })) |dep| {
         step.root_module.addImport("z2d", dep.module("z2d"));
     }
+    if (step.kind == .@"test") {
+        if (b.lazyDependency("ziglyph", .{
+            .target = step.root_module.resolved_target.?,
+            .optimize = step.root_module.optimize.?,
+        })) |dep| {
+            step.root_module.addImport("ziglyph", dep.module("ziglyph"));
+        }
+    }
     if (b.lazyDependency("uucode", .{
         .target = target,
         .optimize = optimize,