diff --git a/modules/localgov_subsites_paragraphs/localgov_subsites_paragraphs.module b/modules/localgov_subsites_paragraphs/localgov_subsites_paragraphs.module
index 741c03e..c38d12e 100644
--- a/modules/localgov_subsites_paragraphs/localgov_subsites_paragraphs.module
+++ b/modules/localgov_subsites_paragraphs/localgov_subsites_paragraphs.module
@@ -143,7 +143,7 @@ function localgov_subsites_paragraphs_preprocess_paragraph(&$variables) {
       $button = $paragraph->get('localgov_button')->first()->getValue();
       $variables['button_url'] = Url::fromUri($button['uri']);
     }
-    if (!$paragraph->get('localgov_background_image')->isEmpty()) {
+    if ($paragraph->get('localgov_background_image')->entity) {
       $fid = $paragraph->get('localgov_background_image')->entity->field_media_image[0]->getValue()['target_id'];
       $file_url = File::load($fid)->getFileUri();