diff --git a/modules/features/zsh/config.toml b/modules/features/zsh/config.toml
index 7c6eabd..268f8dd 100644
--- a/modules/features/zsh/config.toml
+++ b/modules/features/zsh/config.toml
@@ -37,7 +37,7 @@ template = '''
 	[[blocks.segments]] # Hostname
   type = 'text'
   style = 'plain'
-  template = '{{ if eq .HostName "server02" }}<green>󰒋 {{.HostName}}</> <#7f849c>|</> {{ end }}'
+  template = '{{ if .Env.SSH_CLIENT }}<green>󰒋 {{.HostName}}</> <#7f849c>|</> {{ end }}'
 
 	[[blocks.segments]] # Python venv
   type = 'text'