diff --git a/cv4/main.c b/cv4/main.c
index 2b056f6..ee71136 100644
--- a/cv4/main.c
+++ b/cv4/main.c
@@ -1,4 +1,4 @@
-#include "a_train.c"
+#include "a_train.h"
 #include <stdio.h>
 #include <string.h>