diff --git a/oz-init/init.go b/oz-init/init.go
index a96476c..e0d7fa7 100644
--- a/oz-init/init.go
+++ b/oz-init/init.go
@@ -310,6 +310,9 @@ func (st *initState) handleRunShell(rs *RunShellMsg, msg *ipc.Message) error {
 			cmd.Env = append(cmd.Env, "HOME="+homedir)
 		}
 	}
+	if st.profile.XServer.Enabled {
+		cmd.Env = append(cmd.Env, "DISPLAY=:" + strconv.Itoa(st.display))
+	}
 	cmd.Env = append(cmd.Env, "PATH=/usr/bin:/bin")
 	cmd.Env = append(cmd.Env, fmt.Sprintf("PS1=[%s] $ ", st.profile.Name))
 	st.log.Info("Executing shell...")