diff --git a/src/lib/train/train-detail.svelte b/src/lib/train/train-detail.svelte
index b765ce2..0194d08 100644
--- a/src/lib/train/train-detail.svelte
+++ b/src/lib/train/train-detail.svelte
@@ -61,7 +61,7 @@
{:else}
-
+
{#if serviceDetail.pis}
{/if}
diff --git a/src/lib/train/trainIcons.svelte b/src/lib/train/trainIcons.svelte
index 53846ad..adbc585 100644
--- a/src/lib/train/trainIcons.svelte
+++ b/src/lib/train/trainIcons.svelte
@@ -1,35 +1,33 @@
-{#if firstClass}
+{#if serviceDetails.firstClass}
{/if}
-{#if hasCatering}
+{#if serviceDetails.catering}
{/if}
-{#if sleeper}
+{#if serviceDetails.sleeper}
{/if}
-{#if vstp}
+{#if serviceDetails.vstp}