diff --git a/tools/templates/files/fileTemplates/Template Presentation Classes.kt.child.2.kt b/tools/templates/files/fileTemplates/Template Presentation Classes.kt.child.2.kt
index b19f3e728e..0080f8d905 100644
--- a/tools/templates/files/fileTemplates/Template Presentation Classes.kt.child.2.kt	
+++ b/tools/templates/files/fileTemplates/Template Presentation Classes.kt.child.2.kt	
@@ -7,8 +7,8 @@ import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.tooling.preview.PreviewParameter
-import io.element.android.libraries.designsystem.preview.ElementPreviewDark
-import io.element.android.libraries.designsystem.preview.ElementPreviewLight
+import io.element.android.libraries.designsystem.preview.ElementPreview
+import io.element.android.libraries.designsystem.preview.PreviewsDayNight
 import io.element.android.libraries.designsystem.theme.components.Text
 
 @Composable
@@ -24,19 +24,12 @@ fun ${NAME}View(
     }
 }
 
-@Preview
+@PreviewsDayNight
 @Composable
-fun ${NAME}ViewLightPreview(@PreviewParameter(${NAME}StateProvider::class) state: ${NAME}State) =
-    ElementPreviewLight { ContentToPreview(state) }
-
-@Preview
-@Composable
-fun ${NAME}ViewDarkPreview(@PreviewParameter(${NAME}StateProvider::class) state: ${NAME}State) =
-    ElementPreviewDark { ContentToPreview(state) }
-
-@Composable
-private fun ContentToPreview(state: ${NAME}State) {
+internal fun ${NAME}ViewPreview(
+    @PreviewParameter(${NAME}StateProvider::class) state: ${NAME}State
+) = ElementPreview {
     ${NAME}View(
         state = state,
     )
-}
\ No newline at end of file
+}