diff --git a/app/agent/models/agent.js b/app/agent/models/agent.js
index 54866ef90..c41726511 100644
--- a/app/agent/models/agent.js
+++ b/app/agent/models/agent.js
@@ -1,5 +1,7 @@
-export function AgentViewModel(data) {
-  this.IPAddress = data.IPAddress;
-  this.NodeName = data.NodeName;
-  this.NodeRole = data.NodeRole;
+export class AgentViewModel {
+  constructor(data) {
+    this.IPAddress = data.IPAddress;
+    this.NodeName = data.NodeName;
+    this.NodeRole = data.NodeRole;
+  }
 }