1 files changed, 16 insertions, 2 deletions
diff --git a/bindings/objc/test/SwordModuleLongRunTest.mm b/bindings/objc/test/SwordModuleLongRunTest.mm
index 3ed363d..699a18c 100644
--- a/bindings/objc/test/SwordModuleLongRunTest.mm
+++ b/bindings/objc/test/SwordModuleLongRunTest.mm
@@ -7,12 +7,26 @@
 //
 
 #import "SwordModuleTest.h"
+#ifdef TARGET_IPHONE_SIMULATOR
+#import "SwordManager.h"
+#import "Configuration.h"
+#import "iOSConfiguration.h"
+#import "SwordModule.h"
+#import "SwordBibleTextEntry.h"
+#import "VerseEnumerator.h"
+#import "SwordListKey.h"
+#else
 #import "ObjCSword/ObjCSword.h"
+#endif
 
 @implementation SwordModuleTest
 
 - (void)setUp {
+#ifdef TARGET_IPHONE_SIMULATOR
+    [[Configuration config] setClass:[iOSConfiguration class]];
+#else
     [[Configuration config] setClass:[OSXConfiguration class]];
+#endif
     mod = [[SwordManager defaultManager] moduleWithName:@"GerNeUe"];    
 }
 
@@ -52,7 +66,7 @@
     
     SwordListKey *lk = [SwordListKey listKeyWithRef:@"gen 1:1-2"];
     [lk setPersist:YES];
-    [com setKey:lk];
+    [com setSwordKey:lk];
     NSString *ref = nil;
     NSString *rendered = nil;
     int count = 0;
@@ -77,7 +91,7 @@
     int count = 0;
     while(![lk error]) {
         ref = [lk keyText];
-        [com setKey:lk];
+        [com setSwordKey:lk];
         rendered = [com renderedText];
         NSLog(@"%@:%@", ref, rendered);
         [lk increment];