diff --git a/src/test.cpp b/src/test.cpp
index 795d845d..8d1e380c 100644
--- a/src/test.cpp
+++ b/src/test.cpp
@@ -1,6 +1,12 @@
 #include <iostream>
 #include <ctime>
 
+#include <cuda_runtime.h>
+#include <device_launch_parameters.h>
+
+#include <cublas_v2.h>
+#include <cudnn.h>
+
 #include "marian.h"
 
 using namespace marian;