From 1992cdde2e4f9c640cc412232a2884c19a042a2a Mon Sep 17 00:00:00 2001
From: Christopher Rogers <christorogers@gmail.com>
Date: Tue, 4 Oct 2016 12:38:20 +0900
Subject: [PATCH] Swift 3 NSData to Data

---
 mogenerator.m | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/mogenerator.m b/mogenerator.m
index b56e050f..d9915d17 100644
--- a/mogenerator.m
+++ b/mogenerator.m
@@ -537,11 +537,14 @@ - (NSString*)objectAttributeClassName {
     } else {
         result = [self attributeValueClassName];
     }
-    if (gSwift && [result isEqualToString:@"NSString"]) {
-        result = @"String";
-    }
-    if (gSwift && [result isEqualToString:@"NSDate"]) {
-        result = @"Date";
+    if (gSwift) {
+        if ([result isEqualToString:@"NSString"]) {
+            result = @"String";
+        } else if ([result isEqualToString:@"NSDate"]) {
+            result = @"Date";
+        } else if ([result isEqualToString:@"NSData"]) {
+            result = @"Data";
+        }
     }
     return result;
 }