1 files changed, 8 insertions, 6 deletions
diff --git a/build.zig b/build.zig
index 572a55c..79d8d8f 100644
--- a/build.zig
+++ b/build.zig
@@ -18,13 +18,13 @@ pub fn build(b: *std.Build) void {
 
     const db = b.addModule("db", .{
         .root_source_file = b.path("src/db.zig"),
+        .target = target,
+        .optimize = optimize,
     });
     db.addImport("lmdb", lmdb);
 
     const lmdb_tests = b.addTest(.{
-        .root_source_file = b.path("src/lmdb.zig"),
-        .target = target,
-        .optimize = optimize,
+        .root_module = lmdb,
     });
     lmdb_tests.addIncludePath(b.path("lmdb"));
     lmdb_tests.addCSourceFiles(.{ .files = &.{
@@ -34,9 +34,7 @@ pub fn build(b: *std.Build) void {
     lmdb_tests.linkLibC();
 
     const db_tests = b.addTest(.{
-        .root_source_file = b.path("src/db.zig"),
-        .target = target,
-        .optimize = optimize,
+        .root_module = db,
     });
     db_tests.root_module.addImport("lmdb", lmdb);
 
@@ -54,4 +52,8 @@ pub fn build(b: *std.Build) void {
     test_step.dependOn(&db_tests.step);
     test_step.dependOn(&lmdb_test_bin.step);
     test_step.dependOn(&db_test_bin.step);
+
+    const check = b.step("check", "Check if ziglmdb compiles");
+    check.dependOn(&lmdb_tests.step);
+    check.dependOn(&db_tests.step);
 }