diff --git a/src/iGeodeLib.hpp b/src/iGeodeLib.hpp
index 7422073..14a9f5a 100644
--- a/src/iGeodeLib.hpp
+++ b/src/iGeodeLib.hpp
@@ -3,8 +3,7 @@
 
 #pragma once
 #include <functional>
-
-@class NSString;
+#import <UIKit/UIKit.h>
 
 void showAlert(const char *title, const char *message, const char *Btn);
 NSString* iOSVersion();
diff --git a/src/iGeodeLib.mm b/src/iGeodeLib.mm
index f4a390e..d5835f6 100644
--- a/src/iGeodeLib.mm
+++ b/src/iGeodeLib.mm
@@ -1,4 +1,3 @@
-#import <UIKit/UIKit.h>
 #include <functional>
 #include "iGeodeLib.hpp"