diff --git a/build/azure-devops/templates/tests/run-integration-tests.yml b/build/azure-devops/templates/tests/run-integration-tests.yml
index 92ee7ba22..c1b3a05ca 100644
--- a/build/azure-devops/templates/tests/run-integration-tests.yml
+++ b/build/azure-devops/templates/tests/run-integration-tests.yml
@@ -26,5 +26,8 @@ steps:
   displayName: 'Run Integration Tests'
   inputs:
     command: test
+    includeNuGetOrg: true
     projects: 'src/Promitor.Tests.Integration/Promitor.Tests.Integration.csproj'
+    feedsToUse: 'config'
+    nugetConfigPath: 'src/NuGet.config'
     arguments: '--configuration ${{ parameters.buildConfiguration }} --filter "Agent=${{ parameters.agentName}}"'
diff --git a/build/azure-devops/templates/tests/run-unit-tests.yml b/build/azure-devops/templates/tests/run-unit-tests.yml
index 8deaf39f6..ce7f33955 100644
--- a/build/azure-devops/templates/tests/run-unit-tests.yml
+++ b/build/azure-devops/templates/tests/run-unit-tests.yml
@@ -9,8 +9,11 @@ steps:
   parameters:
     dotnetVersion: '${{ parameters.dotnetVersion }}'
 - task: DotNetCoreCLI@2
-  displayName: 'dotnet test'
+  displayName: 'Run unit tests'
   inputs:
     command: test
+    includeNuGetOrg: true
     projects: 'src/Promitor.Tests.Unit/Promitor.Tests.Unit.csproj'
+    feedsToUse: 'config'
+    nugetConfigPath: 'src/NuGet.config'
     arguments: '--configuration ${{ parameters.buildConfiguration }}'
diff --git a/src/NuGet.config b/src/NuGet.config
new file mode 100644
index 000000000..2b317ba0a
--- /dev/null
+++ b/src/NuGet.config
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="utf-8"?>
+<configuration>
+    <activePackageSource>
+        <add key="NuGet.org" value="https://api.nuget.org/v3/index.json" />
+    </activePackageSource>
+</configuration>
diff --git a/src/docker-compose.yml b/src/docker-compose.yml
index f6f33ec9c..af65fd02e 100644
--- a/src/docker-compose.yml
+++ b/src/docker-compose.yml
@@ -56,4 +56,4 @@ services:
       - ./../config/grafana/datasources.yml:/etc/grafana/provisioning/datasources/promitor.yml
       - ./../config/grafana/dashboards.yml:/etc/grafana/provisioning/dashboards/promitor.yml
       - ./../config/grafana/dashboard-config.json:/var/lib/grafana/dashboards/promitor.json
-      - ./../config/grafana/grafana.ini:/etc/grafana/grafana.ini
+      - ./../config/grafana/grafana.ini:/etc/grafana/grafana.ini
\ No newline at end of file