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