From 3a19a177a7b506f1dc45d02eedefd26cf16e9d4f Mon Sep 17 00:00:00 2001
From: NRZ Code <nrzcode@protonmail.com>
Date: Wed, 21 Aug 2024 01:12:27 -0300
Subject: [PATCH] Preserve underscore variable ($_)

---
 bash_command_timer.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bash_command_timer.sh b/bash_command_timer.sh
index 625b9dd..8e3dd56 100644
--- a/bash_command_timer.sh
+++ b/bash_command_timer.sh
@@ -227,8 +227,8 @@ function BCTRegisterCallbacksWithBashPreexec() {
   precmd_functions+=(BCTPostCommand)
 }
 function BCTRegisterCallbacksDirectly() {
-  trap 'BCTPreCommand' DEBUG
-  PROMPT_COMMAND='BCTPostCommand'
+  trap 'BCTPreCommand "$_"' DEBUG
+  PROMPT_COMMAND+=('BCTPostCommand')
 }
 # Case 1: User-supplied path via BASH_PREEXEC_LOCATION
 if ! [ -z "$BASH_PREEXEC_LOCATION" ] && [ -f "$BASH_PREEXEC_LOCATION" ]; then