diff --git a/Memory/Memory.hpp b/Memory/Memory.hpp
index e880a04..cb9d8fe 100644
--- a/Memory/Memory.hpp
+++ b/Memory/Memory.hpp
@@ -1,5 +1,6 @@
 #pragma once
-
+#include <windows.h>
+#include <psapi.h>
 #include <spdlog/spdlog.h>
 
 #define AUTO_ASSEMBLE_TRAMPOLINE(ADDRESS, TRAMPOLINE_LENGTH, INSTRUCTIONS) \
@@ -78,7 +79,7 @@ class Memory
 		 */
 		static PVOID SetupOrClearHardwareBreakPointForAllThreads(uintptr_t targetAddress, PVOID vehHandle, bool enable, PVECTORED_EXCEPTION_HANDLER pVEH = nullptr, int hwIndex = 0);
 	private:
-		static bool WaitForModule(const std::string& module_name, int timeoutMs, int intervalMs);
+		static MODULEINFO WaitForModule(const std::string& module_name, int timeoutMs = 15000, int intervalMs = 500);
 		struct PatchInfo {
 			void* address;
 			std::vector<BYTE> originalBytes;