Update tag.rb

rapid7 · Jun 2, 2014 · c98de1d · c98de1d
1 parent c537310
commit c98de1d
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/lib/nexpose/tag.rb b/lib/nexpose/tag.rb
@@ -212,12 +212,13 @@ def initialize(name, type, id = -1)
 
     # Set the color but validate it
     def color=(hex)
-      unless Type::Color::constants.include?(hex.to_s.downcase)
-        raise ArgumentError, "Unable to set color to an invalid color.\nUse one of #{Type::Color::constants}"
-      end
+      valid_colors = Type::Color::constants.map { |c| Type::Color.const_get(c) }
+      unless valid_colors.include?(hex.to_s.downcase)
+        raise ArgumentError, "Unable to set color to an invalid color.\nUse one of #{valid_colors}"
+      end 
 
-      @color = hex
-    end
+      @color = hex 
+    end 
 
     # Creates and saves a tag to Nexpose console
     #