handle edge cases

1 files changed, 5 insertions, 1 deletions

diff --git a/src/cwd.rs b/src/cwd.rs
index 7672981..95b6659 100644
--- a/src/cwd.rs
+++ b/src/cwd.rs
@@ -8,7 +8,11 @@ pub fn cwd() -> colored::ColoredString {
     let tilde_expand = env::var("EXPAND_TILDE").unwrap_or("0".into());
 
     match tilde_expand.as_ref() {
-        "0" => path = path.replace(&home[..], "~"),
+        "0" => {
+            if (&path[..]).starts_with(&home[..]) && home.len() > 0 {
+                path = path.replacen(&home[..], "~", 1);
+            }
+        }
         _ => {}
     };