thodg/got

diff --git a/lib/worktree.c b/lib/worktree.c
index 1c36cc6..e8828ca 100644
--- a/lib/worktree.c
+++ b/lib/worktree.c
@@ -4410,13 +4410,12 @@ revert_file(void *arg, unsigned char status, unsigned char staged_status,
 		if (err)
 			goto done;
 
-		te_name = basename(ie->path);
-		if (te_name == NULL) {
-			err = got_error_from_errno2("basename", ie->path);
+		err = got_path_basename(&te_name, ie->path);
+		if (err)
 			goto done;
-		}
 
 		te = got_object_tree_find_entry(tree, te_name);
+		free(te_name);
 		if (te == NULL && status != GOT_STATUS_ADD &&
 		    staged_status != GOT_STATUS_ADD) {
 			err = got_error_path(ie->path, GOT_ERR_NO_TREE_ENTRY);