diff --git a/management/cmd/management.go b/management/cmd/management.go
index b2053147aee..0fc5d106792 100644
--- a/management/cmd/management.go
+++ b/management/cmd/management.go
@@ -81,7 +81,7 @@ var (
 		PreRunE: func(cmd *cobra.Command, args []string) error {
 			flag.Parse()
 
-			ctx := context.WithValue(cmd.Context(), formatter.LogSourceKey, formatter.SystemSource)
+			ctx := context.WithValue(cmd.Context(), formatter.ExecutionContextKey, formatter.SystemSource)
 
 			err := util.InitLog(logLevel, logFile)
 			if err != nil {
diff --git a/management/cmd/migration_up.go b/management/cmd/migration_up.go
index 5d1c0ed36d1..7aa11f0c92c 100644
--- a/management/cmd/migration_up.go
+++ b/management/cmd/migration_up.go
@@ -30,7 +30,7 @@ var upCmd = &cobra.Command{
 		}
 
 		//nolint
-		ctx := context.WithValue(cmd.Context(), formatter.LogSourceKey, formatter.SystemSource)
+		ctx := context.WithValue(cmd.Context(), formatter.ExecutionContextKey, formatter.SystemSource)
 
 		if err := server.MigrateFileStoreToSqlite(ctx, mgmtDataDir); err != nil {
 			return err