lib/packetgen/header/dot11/management.rb in packetgen-3.2.0 vs lib/packetgen/header/dot11/management.rb in packetgen-3.2.1

@@ -51,15 +51,12 @@
         # @param [Integer,String] type element type
         # @param [Object] value element value
         # @return [self]
         # @since 2.1.3
         def add_element(type:, value:)
-          if self[:body].is_a? SubMngt
-            self[:body].elements << { type: type, value: value }
-          else
-            raise FormatError, 'Before adding an Element, you have to add a Dot11::SubMngt subclass instance'
-          end
+          raise FormatError, 'Before adding an Element, you have to add a Dot11::SubMngt subclass instance' unless self[:body].is_a? SubMngt
 
+          self[:body].elements << { type: type, value: value }
           self
         end
 
         # Invert mac1 and mac2 (resp. destination address and source address).
         # @return [self]