diff --git a/build.zig b/build.zig
index d1ffbfc..7ee4646 100644
--- a/build.zig
+++ b/build.zig
@@ -11,13 +11,13 @@ pub fn build(b: *std.Build) void {
 
     // 1. Define the module so other projects can import it
     const mod = b.addModule("dimal", .{
-        .root_source_file = b.path("src/main.zig"),
+        .root_source_file = b.path("src/lib.zig"),
     });
     mod.addImport("gpu", zig_wgpu.module("zig-wgpu"));
 
     const exe_tests = b.addTest(.{
         .root_module = b.createModule(.{
-            .root_source_file = b.path("src/main.zig"),
+            .root_source_file = b.path("src/test.zig"),
             .target = target,
             .optimize = optimize,
         }),
diff --git a/src/main.zig b/src/lib.zig
similarity index 60%
rename from src/main.zig
rename to src/lib.zig
index 3f760ec..6e494a2 100644
--- a/src/main.zig
+++ b/src/lib.zig
@@ -5,11 +5,3 @@ pub const Dimensions = @import("Dimensions.zig");
 pub const Scales = @import("Scales.zig");
 pub const Base = @import("Base.zig");
 pub const UnitParser = @import("UnitParser.zig");
-
-test {
-    _ = @import("Tensor.zig");
-    _ = @import("Dimensions.zig");
-    _ = @import("Scales.zig");
-    _ = @import("Base.zig");
-    _ = @import("UnitParser.zig");
-}
diff --git a/src/test.zig b/src/test.zig
new file mode 100644
index 0000000..61a47f3
--- /dev/null
+++ b/src/test.zig
@@ -0,0 +1,7 @@
+test {
+    _ = @import("Tensor.zig");
+    _ = @import("Dimensions.zig");
+    _ = @import("Scales.zig");
+    _ = @import("Base.zig");
+    _ = @import("UnitParser.zig");
+}