1 files changed, 6 insertions, 3 deletions
diff --git a/animism-align/frontend/app/utils/annotation.utils.js b/animism-align/frontend/app/utils/annotation.utils.js
index 0884f62..4dd7478 100644
--- a/animism-align/frontend/app/utils/annotation.utils.js
+++ b/animism-align/frontend/app/utils/annotation.utils.js
@@ -16,7 +16,10 @@ export const annotationFadeTimings = annotation => {
   }
 }
 
-export const thumbnailURL = data => {
-  if (data.type === 'video') return data.settings.video.thumbnail_url
-  if (data.type === 'image') return data.settings.thumbnail.url
+export const thumbnailURL = media => {
+  switch (media.type) {
+    case 'video': return media.settings.video.thumbnail_url
+    case 'image': return media.settings.thumbnail.url
+    default:      return null
+  }
 }