diff --git a/main.cpp b/main.cpp
index a4913fb..56cbca0 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1028,11 +1028,12 @@ int main(int argc, const char** argv) {
 
 	try {
 
-		output_handle = GetStdHandle(STD_OUTPUT_HANDLE);
-
 		if (is_injected) {
 
-			AttachConsole(ATTACH_PARENT_PROCESS);
+			if(!AttachConsole(ATTACH_PARENT_PROCESS)) {
+				fatal_error("failed to attach to parent");
+			}
+			output_handle = GetStdHandle(STD_OUTPUT_HANDLE);
 
 			log("attached\n");
 
@@ -1066,6 +1067,7 @@ int main(int argc, const char** argv) {
 			ExitThread(0);
 
 		} else {
+			output_handle = GetStdHandle(STD_OUTPUT_HANDLE);
 
 			int r = parse_args(argc, argv);
 			if (r) return r;
diff --git a/sc_hook.cpp b/sc_hook.cpp
index c33cc66..09cbbbf 100644
--- a/sc_hook.cpp
+++ b/sc_hook.cpp
@@ -1,4 +1,5 @@
 #include <cstdint>
+#include <stdexcept>
 
 #include "codegen.h"
 #include "x86dec.h"