diff --git a/colorbox.inc b/colorbox.inc
index b8d0aaf..cbdc347 100644
--- a/colorbox.inc
+++ b/colorbox.inc
@@ -8,6 +8,7 @@
 use Drupal\Core\Template\Attribute;
 use Drupal\Component\Utility\Crypt;
 use Drupal\Component\Utility\Unicode;
+use Drupal\image\Entity\ImageStyle;
 
 /**
  * Prepares variables for colorbox formatter templates.
@@ -146,7 +147,7 @@ function template_preprocess_colorbox_formatter(&$variables) {
   $variables['image']['#uri'] = empty($item->uri) ? $image_uri : $item->uri;
 
   if (!empty($settings['colorbox_image_style'])) {
-    $style = entity_load('image_style', $settings['colorbox_image_style']);
+    $style = ImageStyle::load($settings['colorbox_image_style']);
     $variables['url'] = $style->buildUrl($image_uri);
   }
   else {
diff --git a/src/Plugin/Field/FieldFormatter/ColorboxFormatter.php b/src/Plugin/Field/FieldFormatter/ColorboxFormatter.php
index 5848cad..9c7f195 100644
--- a/src/Plugin/Field/FieldFormatter/ColorboxFormatter.php
+++ b/src/Plugin/Field/FieldFormatter/ColorboxFormatter.php
@@ -7,6 +7,7 @@
 
 namespace Drupal\colorbox\Plugin\Field\FieldFormatter;
 
+use Drupal\image\Entity\ImageStyle;
 use Drupal\image\Plugin\Field\FieldFormatter\ImageFormatterBase;
 use Drupal\Core\Field\FieldItemListInterface;
 use Drupal\Core\Form\FormStateInterface;
@@ -221,12 +222,12 @@ class ColorboxFormatter extends ImageFormatterBase {
     // Collect cache tags to be added for each item in the field.
     $cache_tags = array();
     if (!empty($settings['colorbox_node_style'])) {
-      $image_style = entity_load('image_style', $settings['colorbox_node_style']);
+      $image_style = ImageStyle::load($settings['colorbox_node_style']);
       $cache_tags = $image_style->getCacheTags();
     }
     $cache_tags_first = array();
     if (!empty($settings['colorbox_node_style_first'])) {
-      $image_style_first = entity_load('image_style', $settings['colorbox_node_style_first']);
+      $image_style_first = ImageStyle::load($settings['colorbox_node_style_first']);
       $cache_tags_first = $image_style_first->getCacheTags();
     }