diff --git a/templates/components/shift_button.rs.html b/templates/components/shift_button.rs.html
new file mode 100644
index 0000000..0089cdd
--- /dev/null
+++ b/templates/components/shift_button.rs.html
@@ -0,0 +1,11 @@
+@use super::close_shift_button_html;
+@use super::open_shift_button_html;
+@use cm_lib::models::Shift;
+
+@(open_shift: Option<Shift>)
+
+@if open_shift.is_none() {
+  @:open_shift_button_html()
+} else {
+  @:close_shift_button_html(open_shift.unwrap())
+}
diff --git a/templates/home.rs.html b/templates/home.rs.html
index da6e090..1ca3a27 100644
--- a/templates/home.rs.html
+++ b/templates/home.rs.html
@@ -1,6 +1,5 @@
 @use super::base_html;
-@use super::components::close_shift_button_html;
-@use super::components::open_shift_button_html;
+@use super::components::shift_button_html;
 @use cm_lib::models::Shift;
 
 @(open_shift: Option<Shift>)
@@ -9,11 +8,7 @@
 
 <main>
   <div>
-    @if open_shift.is_none() {
-      @:open_shift_button_html()
-    } else {
-      @:close_shift_button_html(open_shift.unwrap())
-    }
+    @:shift_button_html(open_shift)
   </div>
 </main>