diff --git a/.gitignore b/.gitignore
index ebeeab3..6cbb503 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
+.idea/*
 cmake-build-debug/*
diff --git a/exploit/exploit.c b/exploit/exploit.c
index 42762d3..6d8fbff 100644
--- a/exploit/exploit.c
+++ b/exploit/exploit.c
@@ -108,7 +108,10 @@ int check_function(struct libusb_device_bundle *bundle)
 
 int exploit_device()
 {
-    int ret = complete_stage(stage1_function);
+    int ret = complete_stage(check_function);
+    if(ret == 0) return ret;
+
+    ret = complete_stage(stage1_function);
     if(ret == 0)
     {
         ret = complete_stage(stage2_function);