diff --git a/FCModel/FCModel.h b/FCModel/FCModel.h
index 8e0f9ff..c8f6cf6 100644
--- a/FCModel/FCModel.h
+++ b/FCModel/FCModel.h
@@ -8,7 +8,7 @@
 #import <Foundation/Foundation.h>
 #include <AvailabilityMacros.h>
 
-#ifdef COCOAPODS
+#ifdef FMDB_FRAMEWORK
 #import <FMDB/FMDatabase.h>
 #else
 #import "FMDatabase.h"
diff --git a/FCModel/FCModel.m b/FCModel/FCModel.m
index 19cd458..042f665 100644
--- a/FCModel/FCModel.m
+++ b/FCModel/FCModel.m
@@ -10,8 +10,15 @@
 #import "FCModel.h"
 #import "FCModelCachedObject.h"
 #import "FCModelDatabaseQueue.h"
-#import "FMDatabase.h"
-#import "FMDatabaseAdditions.h"
+
+#ifdef FMDB_FRAMEWORK
+    #import <FMDB/FMDatabase.h>
+    #import <FMDB/FMDatabaseAdditions.h>
+#else
+    #import "FMDatabase.h"
+    #import "FMDatabaseAdditions.h"
+#endif
+
 #import <sqlite3.h>
 #import <Security/Security.h>
 
diff --git a/FCModel/FCModelDatabaseQueue.h b/FCModel/FCModelDatabaseQueue.h
index ca278e8..e382a73 100644
--- a/FCModel/FCModelDatabaseQueue.h
+++ b/FCModel/FCModelDatabaseQueue.h
@@ -7,7 +7,7 @@
 
 #import <Foundation/Foundation.h>
 
-#ifdef COCOAPODS
+#ifdef FMDB_FRAMEWORK
 #import <FMDB/FMDatabase.h>
 #else
 #import "FMDatabase.h"