Added missing generics

1 files changed, 4 insertions, 4 deletions

diff --git a/SmartDeviceLink/SDLCreateInteractionChoiceSet.m b/SmartDeviceLink/SDLCreateInteractionChoiceSet.m
index adb785c52..3d443d719 100644
--- a/SmartDeviceLink/SDLCreateInteractionChoiceSet.m
+++ b/SmartDeviceLink/SDLCreateInteractionChoiceSet.m
@@ -27,7 +27,7 @@
     return [parameters objectForKey:SDLNameInteractionChoiceSetId];
 }
 
-- (void)setChoiceSet:(NSMutableArray *)choiceSet {
+- (void)setChoiceSet:(NSMutableArray<SDLChoice *> *)choiceSet {
     if (choiceSet != nil) {
         [parameters setObject:choiceSet forKey:SDLNameChoiceSet];
     } else {
@@ -35,12 +35,12 @@
     }
 }
 
-- (NSMutableArray *)choiceSet {
-    NSMutableArray *array = [parameters objectForKey:SDLNameChoiceSet];
+- (NSMutableArray<SDLChoice *> *)choiceSet {
+    NSMutableArray<SDLChoice *> *array = [parameters objectForKey:SDLNameChoiceSet];
     if ([array count] < 1 || [[array objectAtIndex:0] isKindOfClass:SDLChoice.class]) {
         return array;
     } else {
-        NSMutableArray *newList = [NSMutableArray arrayWithCapacity:[array count]];
+        NSMutableArray<SDLChoice *> *newList = [NSMutableArray arrayWithCapacity:[array count]];
         for (NSDictionary<NSString *, id> *dict in array) {
             [newList addObject:[[SDLChoice alloc] initWithDictionary:(NSDictionary *)dict]];
         }