From a7fdbe502bbbeafe28dc2376f688cb255aa7e8a7 Mon Sep 17 00:00:00 2001
From: Tudor Stanciu <tudor.stanciu94@gmail.com>
Date: Wed, 23 Dec 2020 01:43:12 +0200
Subject: [PATCH] small fixes

---
 NetworkResurrector.Agent.Application/Queries/GetMachines.cs | 4 ++--
 NetworkResurrector.Agent/Startup.cs                         | 4 ++++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/NetworkResurrector.Agent.Application/Queries/GetMachines.cs b/NetworkResurrector.Agent.Application/Queries/GetMachines.cs
index f102530..e453685 100644
--- a/NetworkResurrector.Agent.Application/Queries/GetMachines.cs
+++ b/NetworkResurrector.Agent.Application/Queries/GetMachines.cs
@@ -37,8 +37,8 @@ namespace NetworkResurrector.Agent.Application.Queries
 
             public async Task<Model[]> Handle(Query request, CancellationToken cancellationToken)
             {
-                var bots = await _repository.GetMachines();
-                var result = _mapper.Map<Model[]>(bots);
+                var machines = await _repository.GetMachines();
+                var result = _mapper.Map<Model[]>(machines);
 
                 return result;
             }
diff --git a/NetworkResurrector.Agent/Startup.cs b/NetworkResurrector.Agent/Startup.cs
index 0e74282..48e088e 100644
--- a/NetworkResurrector.Agent/Startup.cs
+++ b/NetworkResurrector.Agent/Startup.cs
@@ -9,6 +9,7 @@ using Microsoft.Extensions.Hosting;
 using NDB.Extensions.Swagger;
 using NDB.Security.Authentication.Identity;
 using NetworkResurrector.Agent.Application;
+using NetworkResurrector.Agent.Domain.Data;
 using Newtonsoft.Json;
 using System.Reflection;
 
@@ -44,6 +45,9 @@ namespace NetworkResurrector.Agent
             // Swagger
             services.AddSwagger("NetworkResurrectorAgent API");
 
+            // Data access
+            services.AddDataAccess();
+
             // Application
             services.AddApplicationServices();
         }