diff --git a/codegen/nuget.config b/codegen/nuget.config
index 6cba8660c..c066f688c 100644
--- a/codegen/nuget.config
+++ b/codegen/nuget.config
@@ -1,6 +1,10 @@
 <?xml version="1.0" encoding="utf-8"?>
 <configuration>
-  <packageSources>
-    <add key="generator" value="https://pkgs.dev.azure.com/azure-sdk/public/_packaging/azure-sdk-for-net/nuget/v3/index.json" />
-  </packageSources>
+  <packageSourceMapping>
+    <packageSource key="azure-sdk-dev-feed">
+      <package pattern="Microsoft.TypeSpec.Generator.ClientModel" />
+      <package pattern="Microsoft.TypeSpec.Generator" />
+      <package pattern="Microsoft.TypeSpec.Generator.Input" />
+    </packageSource>
+  </packageSourceMapping>
 </configuration>
\ No newline at end of file
diff --git a/nuget.config b/nuget.config
index 3da872e48..6bf477c46 100644
--- a/nuget.config
+++ b/nuget.config
@@ -6,4 +6,13 @@
     <add key="nuget" value="https://api.nuget.org/v3/index.json" />
     <add key="azure-sdk-dev-feed" value="https://pkgs.dev.azure.com/azure-sdk/public/_packaging/azure-sdk-for-net/nuget/v3/index.json" />
   </packageSources>
+  <packageSourceMapping>
+    <packageSource key="azure-sdk-dev-feed">
+      <package pattern="System.ClientModel" />
+      <package pattern="Microsoft.ClientModel.TestFramework" />
+    </packageSource>
+    <packageSource key="nuget">
+      <package pattern="*" />
+    </packageSource>
+  </packageSourceMapping>
 </configuration>
\ No newline at end of file