diff --git a/pkg/hayai@.service b/pkg/hayai@.service
index a945975..170cd21 100644
--- a/pkg/hayai@.service
+++ b/pkg/hayai@.service
@@ -1,12 +1,11 @@
 [Unit]
 Description=hayai - An EEW system for Linux
-After=network.target
+PartOf=graphical-session.target
 StartLimitIntervalSec=60
 StartLimitBurst=4
 
 [Service]
 User=%i
-Environment=DISPLAY=:0
 ExecStart=/usr/bin/hayai
 Restart=on-failure
 RestartSec=1
@@ -21,4 +20,4 @@ MemoryDenyWriteExecute=true
 NoNewPrivileges=true
 
 [Install]
-WantedBy=multi-user.target
\ No newline at end of file
+WantedBy=graphical-session.target
\ No newline at end of file