diff --git a/hls/assignment/sobel/main_testbench.cpp b/hls/assignment/sobel/main_testbench.cpp
index 1eb52e1..1609d62 100644
--- a/hls/assignment/sobel/main_testbench.cpp
+++ b/hls/assignment/sobel/main_testbench.cpp
@@ -3,7 +3,7 @@
 #include <string.h>
 
 #include "sobel.h"
-#include "test.h"
+#include "main_testbench.h"
 
 uint8_t output[HEIGHT*WIDTH];