1 files changed, 6 insertions, 1 deletions

diff --git a/usr.sbin/diskpart/diskpart.c b/usr.sbin/diskpart/diskpart.c
index 1c6638dbfe660..932b76a1fdaa1 100644
--- a/usr.sbin/diskpart/diskpart.c
+++ b/usr.sbin/diskpart/diskpart.c
@@ -375,8 +375,13 @@ promptfordisk(name)
 		(void) fgets(buf, BUFSIZ, stdin);
 		if (buf[0] == 0)
 			dp->d_type = 1;
-		else
+		else {
+			int len = strlen(buf);
+
+			if (buf[len - 1] == '\n')
+				buf[len - 1] = '\0';
 			dp->d_type = gettype(buf, dktypenames);
+		}
 		if (dp->d_type >= 0)
 			break;
 		fprintf(stderr, "%s: unrecognized controller type\n", buf);