diff --git a/include/OpenCL.h b/include/OpenCL.h
index 8865467..f2aee09 100644
--- a/include/OpenCL.h
+++ b/include/OpenCL.h
@@ -1,8 +1,16 @@
 #pragma once
 
+#include <string>
+#include <SFML/Graphics.hpp>
+#include <unordered_map>
+#include <iostream>
+#include "Vector4.hpp"
+#include <string.h>
+
 #ifdef linux
 #include <CL/cl.h>
 #include <CL/opencl.h>
+#include <GL/glx.h>
 
 #elif defined _WIN32
 #define CL_USE_DEPRECATED_OPENCL_1_2_APIS
@@ -18,13 +26,6 @@
 #include <OpenCL/opencl.h>
 
 #endif
-
-#include <string>
-#include <SFML/Graphics.hpp>
-#include <unordered_map>
-#include <iostream>
-#include "Vector4.hpp"
-
 struct device {
 	
 	cl_device_id id;
@@ -112,4 +113,4 @@ private:
 
 	bool vr_assert(int error_code, std::string function_name);
 
-};
\ No newline at end of file
+};
diff --git a/src/main.cpp b/src/main.cpp
index d31678e..10bb0dc 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -5,7 +5,6 @@
 #include <chrono>
 #include "util.hpp"
 #include <thread>
-#include <vector>
 #include "OpenCL.h"
 
 float elap_time() {
@@ -154,4 +153,4 @@ int main() {
 	}
 	return 0;
 
-}
\ No newline at end of file
+}