diff --git a/build.psm1 b/build.psm1
index d57e064..30efc9f 100644
--- a/build.psm1
+++ b/build.psm1
@@ -119,22 +119,6 @@ function Start-Build
         Write-Host "`n[Build the AIShell module ...]`n" -ForegroundColor Green
         $aish_module_csproj = GetProjectFile $module_dir
         dotnet publish $aish_module_csproj -c $Configuration -o $module_out_dir
-
-        # Update version for the module manifest file.
-        $projectUrl = 'https://github.com/PowerShell/ProjectMercury'
-        $version = (Get-Item $module_out_dir/AIShell.Integration.dll).VersionInfo.ProductVersion
-        $privateData = "PrivateData = @{ PSData = @{ ProjectUri = '$projectUrl' } }"
-        if ($version -match "(.*)-(.*)") {
-            $version = $matches[1]
-            $prerelease = $matches[2]
-            # Put the prerelease tag in private data.
-            $privateData = "PrivateData = @{ PSData = @{ Prerelease = '$prerelease'; ProjectUri = '$projectUrl' } }"
-        }
-
-        $moduleManifest = Get-Content $module_out_dir/AIShell.psd1 -Raw
-        $moduleManifest = $moduleManifest -replace "ModuleVersion = '.*'", "ModuleVersion = '$version'"
-        $moduleManifest = $moduleManifest -replace "}", "    ${privateData}`n}`n"
-        Set-Content -Path $module_out_dir/AIShell.psd1 -Value $moduleManifest -NoNewline
     }
 
     if ($LASTEXITCODE -eq 0) {
diff --git a/shell/AIShell.Integration/AIShell.psd1 b/shell/AIShell.Integration/AIShell.psd1
index 53f6b11..d2cd2a9 100644
--- a/shell/AIShell.Integration/AIShell.psd1
+++ b/shell/AIShell.Integration/AIShell.psd1
@@ -1,7 +1,7 @@
 @{
     RootModule = 'AIShell.psm1'
     NestedModules = @("AIShell.Integration.dll")
-    ModuleVersion = '0.1.0'
+    ModuleVersion = '1.0.1'
     GUID = 'ECB8BEE0-59B9-4DAE-9D7B-A990B480279A'
     Author = 'Microsoft Corporation'
     CompanyName = 'Microsoft Corporation'
@@ -13,4 +13,5 @@
     VariablesToExport = '*'
     AliasesToExport = @('aish', 'askai', 'fixit')
     HelpInfoURI = 'https://aka.ms/aishell-help'
+    PrivateData = @{ PSData = @{ Prerelease = 'preview1'; ProjectUri = 'https://github.com/PowerShell/ProjectMercury' } }
 }