diff --git a/ai/main.go b/ai/main.go
index a389216..8bd8910 100644
--- a/ai/main.go
+++ b/ai/main.go
@@ -90,8 +90,11 @@ func saveData(word string, keysPressed []utils.InputEvent) {
 	nodeBytes, err := json.Marshal(nodeDatas)
 	edgeBytes, err := json.Marshal(edgeDatas)
 
-	ioutil.WriteFile("nodes.json", nodeBytes, 0644)
-	ioutil.WriteFile("edges.json", edgeBytes, 0644)
+	err = ioutil.WriteFile("nodes.json", nodeBytes, 0644)
+	err = ioutil.WriteFile("edges.json", edgeBytes, 0644)
+	if err != nil {
+		logrus.Error(err)
+	}
 
 }