diff --git a/lib/inspec_tools/csv.rb b/lib/inspec_tools/csv.rb
index d00626a..b1ef09a 100644
--- a/lib/inspec_tools/csv.rb
+++ b/lib/inspec_tools/csv.rb
@@ -74,10 +74,10 @@ def parse_controls(prefix)
         control['tags']['nist'] = nist unless nist.nil? || nist.include?(nil)
         @mapping['control.tags'].each do |tag|
           if tag.first == 'cci'
-            if cci_number.is_a? String
-              control['tags'][tag.first] = [cci_number]
-            else
+            if cci_number.is_a? Array
               control['tags'][tag.first] = cci_number
+            else
+              control['tags'][tag.first] = [cci_number]
             end
             next
           end