Update db_operations/db_write.py

db_operations/db_write.py

@@ -22,8 +22,8 @@ class DBWrite:
         self.collection = PREDICTION_COLLECTION
         if self.db_type == "production":
             self.url = PRODUCTION_URL
-            self.database = DATABASE
-            self.collection = COLLECTION
+            self.database = PRODUCTION_DATABASE
+            self.collection = PRODUCTION_COLLECTION
         
         self.__client = None
         self.__error = 0