From 731c4814a7a2f2fcd472f53abf1e99e86081e52f Mon Sep 17 00:00:00 2001
From: Sen Jiang <JayXon@Gmail.com>
Date: Wed, 4 May 2016 18:18:28 -0700
Subject: [PATCH] Fix #108

---
 bullet-train.zsh-theme | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/bullet-train.zsh-theme b/bullet-train.zsh-theme
index a3da493..54881b5 100644
--- a/bullet-train.zsh-theme
+++ b/bullet-train.zsh-theme
@@ -356,17 +356,21 @@ prompt_context() {
 
 # Prompt previous command execution time
 preexec() {
-    cmd_timestamp=`date +%s`
+  cmd_timestamp=`date +%s`
 }
 
-prompt_cmd_exec_time() {
-  if [[ $BULLETTRAIN_EXEC_TIME_SHOW == false ]]; then
-    return
-  fi
+precmd() {
+  [[ $BULLETTRAIN_EXEC_TIME_SHOW == false ]] && return
 
   local stop=`date +%s`
   local start=${cmd_timestamp:-$stop}
-  let local elapsed=$stop-$start
+  let elapsed=$stop-$start
+  cmd_timestamp=''
+}
+
+prompt_cmd_exec_time() {
+  [[ $BULLETTRAIN_EXEC_TIME_SHOW == false ]] && return
+
   [ $elapsed -gt $BULLETTRAIN_EXEC_TIME_ELAPSED ] && prompt_segment $BULLETTRAIN_EXEC_TIME_BG $BULLETTRAIN_EXEC_TIME_FG "${elapsed}s"
 }