@@ -19,16 +19,16 @@
 #import "OFAutoreleasePool.h"
 #import "OFString.h"
 #import "OFExceptions.h"
 #import "macros.h"
 
-#import "main.h"
+#import "TestsAppDelegate.h"
 
 static OFString *module = @"OFTCPSocket";
 
-void
-tcpsocket_tests()
+@implementation TestsAppDelegate (OFTCPSocketTests)
+- (void)TCPSocketTests
 {
 	OFAutoreleasePool *pool = [[OFAutoreleasePool alloc] init];
 	OFTCPSocket *server, *client = nil, *accepted;
 	OFString *service, *msg;
 	uint16_t port;
@@ -63,5 +63,6 @@
 						     intoBuffer: buf] &&
 	    !memcmp(buf, "Hello!", 6))
 
 	[pool drain];
 }
+@end