diff --git a/src/Plugin/Field/FieldFormatter/PhotoswipeFieldFormatter.php b/src/Plugin/Field/FieldFormatter/PhotoswipeFieldFormatter.php
index 0988f44..557bc38 100644
--- a/src/Plugin/Field/FieldFormatter/PhotoswipeFieldFormatter.php
+++ b/src/Plugin/Field/FieldFormatter/PhotoswipeFieldFormatter.php
@@ -258,6 +258,8 @@ class PhotoswipeFieldFormatter extends FormatterBase {
 
     if (!empty($items)) {
       \Drupal::service('photoswipe.assets_manager')->attach($elements);
+      $elements['#prefix'] = '<div class="photoswipe-gallery">';
+      $elements['#suffix'] = '</div>';
     }
 
     foreach ($items as $delta => $item) {