@@ -16,16 +16,16 @@
 #import "OFArray.h"
 #import "OFAutoreleasePool.h"
 #import "OFString.h"
 #import "OFExceptions.h"
 
-#import "main.h"
+#import "TestsAppDelegate.h"
 
 static OFString *module = @"OFXMLElement";
 
-void
-xmlelement_tests()
+@implementation TestsAppDelegate (OFXMLElementTests)
+- (void)XMLElementTests
 {
 	OFAutoreleasePool *pool = [[OFAutoreleasePool alloc] init];
 	OFXMLElement *elem[2];
 
 	TEST(@"+[elementWithName:]",
@@ -49,5 +49,6 @@
 	    [elem[0] addChild: [OFXMLElement elementWithName: @"bar"]] &&
 	    [[elem[0] string] isEqual: @"<foo foo='b&amp;ar'><bar/></foo>"])
 
 	[pool drain];
 }
+@end