diff --git a/test/Tests.hx b/test/Tests.hx
index 27c1dce..726fdbb 100644
--- a/test/Tests.hx
+++ b/test/Tests.hx
@@ -1,16 +1,5 @@
 package ;
 
-import buddy.BuddySuite;
-import buddy.SuitesRunner;
+import buddy.Buddy;
 
-@:build(buddy.GenerateMain.withSuites([
-    new BaseTest(),
-    new InheritanceTest(),
-    new InlineTest(),
-    new MetaTest(),
-    new SignalTest(),
-    new TestProperty(),
-    new ChainBindTest(),
-    new ExprBindTest(),
-]))
-class Tests extends BuddySuite {}
\ No newline at end of file
+class Tests implements Buddy<[BaseTest, InheritanceTest, InlineTest, MetaTest, SignalTest, TestProperty, ChainBindTest, ExprBindTest]> {}
\ No newline at end of file