From dddb8636577b3dc7d8fc1a9355104afd80115eef Mon Sep 17 00:00:00 2001
From: Matthew Adams <matthew.adams@northscaler.com>
Date: Sat, 2 May 2020 08:43:08 -0500
Subject: [PATCH] change from feedback

---
 themes/powerline-plain/powerline-plain.base.bash  | 6 +-----
 themes/powerline-plain/powerline-plain.theme.bash | 1 +
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/themes/powerline-plain/powerline-plain.base.bash b/themes/powerline-plain/powerline-plain.base.bash
index 1b214fdb..ce91ef89 100644
--- a/themes/powerline-plain/powerline-plain.base.bash
+++ b/themes/powerline-plain/powerline-plain.base.bash
@@ -32,6 +32,7 @@ function __powerline_prompt_command {
   LEFT_PROMPT=""
   SEGMENTS_AT_LEFT=0
   LAST_SEGMENT_COLOR=""
+  PROMPT_AFTER="${POWERLINE_PROMPT_AFTER}"
 
   _save-and-reload-history "${HISTORY_AUTOSAVE:-0}"
 
@@ -51,11 +52,6 @@ function __powerline_prompt_command {
     LEFT_PROMPT+=" "
   fi
 
-  ## after prompt ##
-  if [[ -n "${POWERLINE_PROMPT_AFTER}" ]]; then
-    PROMPT_AFTER="${POWERLINE_PROMPT_AFTER}"
-  fi
-
   PS1="${LEFT_PROMPT}${PROMPT_AFTER}"
 
   ## cleanup ##
diff --git a/themes/powerline-plain/powerline-plain.theme.bash b/themes/powerline-plain/powerline-plain.theme.bash
index 942c411c..709fe341 100644
--- a/themes/powerline-plain/powerline-plain.theme.bash
+++ b/themes/powerline-plain/powerline-plain.theme.bash
@@ -12,6 +12,7 @@ POWERLINE_COMPACT_AFTER_SEPARATOR=${POWERLINE_COMPACT_AFTER_SEPARATOR:=${POWERLI
 POWERLINE_COMPACT_BEFOR_FIRST_SEGMENT=${POWERLINE_COMPACT_BEFORE_FIRST_SEGMENT:=${POWERLINE_COMPACT}}
 POWERLINE_COMPACT_AFTER_LAST_SEGMENT=${POWERLINE_COMPACT_AFTER_LAST_SEGMENT:=${POWERLINE_COMPACT}}
 POWERLINE_COMPACT_PROMPT=${POWERLINE_COMPACT_PROMPT:=${POWERLINE_COMPACT}}
+POWERLINE_PROMPT_AFTER="${POWERLINE_PROMPT_AFTER:-''}"
 
 PYTHON_VENV_CHAR=${POWERLINE_PYTHON_VENV_CHAR:="ⓔ "}
 CONDA_PYTHON_VENV_CHAR=${POWERLINE_CONDA_PYTHON_VENV_CHAR:="ⓔ "}