| @@ -20,9 +20,10 @@ | ||
| 20 | 20 | $css_generator->add_class_styles( |
| 21 | 21 | '{{WRAPPER}}', |
| 22 | 22 | $this->get_wrapper_css( $attributes, '' ), |
| 23 | 23 | $this->get_wrapper_css( $attributes, 'Tablet' ), |
| 24 | - $this->get_wrapper_css( $attributes, 'Mobile' ) | |
| 24 | + $this->get_wrapper_css( $attributes, 'Mobile' ), | |
| 25 | + $css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_wrapper_css( $attributes, $device ); } ) | |
| 25 | 26 | ); |
| 26 | 27 | return $css_generator->generate_css(); |
| 27 | 28 | } |
| 28 | 29 | private function get_wrapper_css( $attributes, $device = '' ) { |
