diff --git a/cv4/a_train.c b/cv4/a_train.c
index 0dd75a7..5d99012 100644
--- a/cv4/a_train.c
+++ b/cv4/a_train.c
@@ -1,4 +1,4 @@
-#include "train.h"
+#include "a_train.h"
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>