diff --git a/src/agent/NetworkResurrector.Agent/Controllers/HostController.cs b/src/agent/NetworkResurrector.Agent/Controllers/HostController.cs
index cfc6668..dee0cc5 100644
--- a/src/agent/NetworkResurrector.Agent/Controllers/HostController.cs
+++ b/src/agent/NetworkResurrector.Agent/Controllers/HostController.cs
@@ -7,7 +7,7 @@ using System.Threading.Tasks;
 namespace NetworkResurrector.Agent.Controllers
 {
     [ApiController]
-  //  [Authorize]
+    [Authorize]
     [Route("host")]
     public class HostController : ControllerBase
     {
diff --git a/src/api/NetworkResurrector.Api.Domain.Data/DbContexts/NetworkDbContext.cs b/src/api/NetworkResurrector.Api.Domain.Data/DbContexts/NetworkDbContext.cs
index 3a422cf..3471159 100644
--- a/src/api/NetworkResurrector.Api.Domain.Data/DbContexts/NetworkDbContext.cs
+++ b/src/api/NetworkResurrector.Api.Domain.Data/DbContexts/NetworkDbContext.cs
@@ -24,6 +24,7 @@ namespace NetworkResurrector.Api.Domain.Data.DbContexts
             modelBuilder.ApplyConfiguration(new MachineConfiguration());
             modelBuilder.ApplyConfiguration(new PowerActionConfiguration());
             modelBuilder.ApplyConfiguration(new PowerActionPerformerConfiguration());
+            modelBuilder.ApplyConfiguration(new PowerActionConfigurationEf());
             modelBuilder.ApplyConfiguration(new MachineAgentConfiguration());
         }
     }
diff --git a/src/server/NetworkResurrector.Server/appsettings.json b/src/server/NetworkResurrector.Server/appsettings.json
index 23c2097..9331376 100644
--- a/src/server/NetworkResurrector.Server/appsettings.json
+++ b/src/server/NetworkResurrector.Server/appsettings.json
@@ -1,6 +1,7 @@
 {
   "urls": "http://*:5062",
   "ConnectionStrings": {
+    //"DatabaseConnection": "***REMOVED***"
     "DatabaseConnection": "***REMOVED***"
   },
   "Logging": {