diff --git a/internal/handler/computer.go b/internal/handler/computer.go
index 313059c..cf130c5 100644
--- a/internal/handler/computer.go
+++ b/internal/handler/computer.go
@@ -26,14 +26,14 @@ func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) {
 	defer conn.Close()
 
 	slog.Info("Websocket connection established!")
-	online := true
+	service.ComputerData.Online = true
 
 	// Read messages
 	for {
 		_, message, err := conn.ReadMessage()
 		if err != nil {
 			slog.Error("WebSocket connection closed by client", slog.Any("error", err))
-			online = false
+			service.ComputerData.Online = false
 			break
 		}
 
@@ -43,7 +43,7 @@ func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) {
 			continue
 		}
 
-		service.AddComputerData(online, clientMessage)
+		service.AddComputerData(clientMessage)
 		slog.Info("Recieved message", slog.Any("message", clientMessage))
 	}
 }
diff --git a/internal/service/computer.go b/internal/service/computer.go
index 84e910d..d6e3462 100644
--- a/internal/service/computer.go
+++ b/internal/service/computer.go
@@ -6,8 +6,7 @@ import (
 
 var ComputerData model.ComputerGraphData
 
-func AddComputerData(online bool, clientMessage model.ComputerWebSocketMessage) {
-	ComputerData.Online = online
+func AddComputerData(clientMessage model.ComputerWebSocketMessage) {
 	ComputerData.Cpu = append(ComputerData.Cpu, int(clientMessage.Cpu))
 	ComputerData.Ram = append(ComputerData.Ram, int(clientMessage.Ram))