@@ -15,11 +15,11 @@
 #import "OFArray.h"
 #import "OFAutoreleasePool.h"
 #import "OFString.h"
 #import "OFExceptions.h"
 
-#import "main.h"
+#import "TestsAppDelegate.h"
 
 static OFString *module = @"OFDictionary";
 static OFString *keys[] = {
 	@"key1",
 	@"key2"
@@ -27,12 +27,12 @@
 static OFString *values[] = {
 	@"value1",
 	@"value2"
 };
 
-void
-dictionary_tests()
+@implementation TestsAppDelegate (OFDictionaryTests)
+- (void)dictionaryTests
 {
 	OFAutoreleasePool *pool = [[OFAutoreleasePool alloc] init];
 	OFMutableDictionary *dict = [OFMutableDictionary dictionary], *dict2;
 	OFEnumerator *key_enum, *obj_enum;
 	OFArray *akeys, *avalues;
@@ -148,5 +148,6 @@
 		     forKey: keys[0]] &&
 	    [dict isEqual: dict2])
 
 	[pool drain];
 }
+@end