diff --git a/pkg/kubeconfig/extensions_test.go b/pkg/kubeconfig/extensions_test.go
index ba899649..2ec50669 100644
--- a/pkg/kubeconfig/extensions_test.go
+++ b/pkg/kubeconfig/extensions_test.go
@@ -263,3 +263,33 @@ func TestCheckContextScopeKubeflexExtensionSetPartialData(t *testing.T) {
 		t.Errorf("Expected %s, got %s", DiagnosisStatusWarning, result)
 	}
 }
+
+func TestCheckExtensionInitialContextNameSetFalse(t *testing.T) {
+	kconf := api.NewConfig()
+	ext := NewRuntimeKubeflexExtension()
+
+	kconf.Extensions = map[string]runtime.Object{
+		ExtensionKubeflexKey: ext,
+	}
+
+	status := CheckExtensionInitialContextNameSet(*kconf)
+	if status != DiagnosisStatusWarning {
+		t.Errorf("Expected %s when ExtensionInitialContextName is not set, got %s", DiagnosisStatusWarning, status)
+	}
+}
+
+func TestCheckExtensionInitialContextNameSetTrue(t *testing.T) {
+	kconf := api.NewConfig()
+
+	ext := NewRuntimeKubeflexExtension()
+	ext.Data[ExtensionInitialContextName] = "kind-kubeflex"
+
+	kconf.Extensions = map[string]runtime.Object{
+		ExtensionKubeflexKey: ext,
+	}
+
+	status := CheckExtensionInitialContextNameSet(*kconf)
+	if status != DiagnosisStatusOK {
+		t.Errorf("Expected %s when ExtensionInitialContextName is set, got %s", DiagnosisStatusOK, status)
+	}
+}