From 25d1ee7b43aa0d38a9a5d7e83d39e573f89f7587 Mon Sep 17 00:00:00 2001
From: bootandy <bootandy@gmail.com>
Date: Wed, 2 May 2018 00:17:40 +0100
Subject: [PATCH] Remove naked unwrap

---
 src/display.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/display.rs b/src/display.rs
index 4a8a068..49131c7 100644
--- a/src/display.rs
+++ b/src/display.rs
@@ -141,8 +141,8 @@ pub fn format_string(
     indentation: &str,
 ) -> String {
     let printable_name = {
-        if short_paths && dir_name.contains('/') {
-            dir_name.split('/').last().unwrap()
+        if short_paths {
+            dir_name.split('/').last().unwrap_or(dir_name)
         } else {
             dir_name
         }