diff --git a/pkg/nvlib/device/mig_profile_test.go b/pkg/nvlib/device/mig_profile_test.go
index 8287455..d1a4f5b 100644
--- a/pkg/nvlib/device/mig_profile_test.go
+++ b/pkg/nvlib/device/mig_profile_test.go
@@ -175,12 +175,6 @@ func TestParseMigProfile(t *testing.T) {
 			false,
 			false,
 		},
-		{
-			"Invalid g.5gb",
-			"1g.5",
-			false,
-			false,
-		},
 		{
 			"Invalid g.5gb",
 			"g.5gb",
@@ -219,49 +213,43 @@ func TestParseMigProfile(t *testing.T) {
 		},
 		{
 			"Invalid 1g.5gb +",
-			"1g.5gb+",
+			"1g.5gb +",
 			false,
 			false,
 		},
 		{
 			"Invalid 1g.5gb+ ",
-			"1g.5gb+",
+			"1g.5gb+ ",
 			false,
 			false,
 		},
 		{
 			"Invalid 1g.5gb+ ,",
-			"1g.5gb+",
-			false,
-			false,
-		},
-		{
-			"Invalid 1g.5gb+,",
-			"1g.5gb+",
+			"1g.5gb+ ,",
 			false,
 			false,
 		},
 		{
 			"Invalid 1g.5gb+,,",
-			"1g.5gb+",
+			"1g.5gb+,,",
 			false,
 			false,
 		},
 		{
 			"Invalid 1g.5gb+me,",
-			"1g.5gb+",
+			"1g.5gb+me,",
 			false,
 			false,
 		},
 		{
 			"Invalid 1g.5gb+me,,",
-			"1g.5gb+",
+			"1g.5gb+me,,",
 			false,
 			false,
 		},
 		{
 			"Invalid 1g.5gb+me, ",
-			"1g.5gb+",
+			"1g.5gb+me, ",
 			false,
 			false,
 		},