diff --git a/emmet-cli/emmet/cli/db.py b/emmet-cli/emmet/cli/db.py
index ce15cb91ea..693511da04 100644
--- a/emmet-cli/emmet/cli/db.py
+++ b/emmet-cli/emmet/cli/db.py
@@ -20,6 +20,7 @@
 
 
 class TaskStore:
+    
     _get_store_from_type: dict[str, Store] = {
         "mongo": MongoStore,
         "s3": S3Store,
@@ -225,7 +226,6 @@ def insert_object(self, *args, **kwargs) -> tuple[int, str]:
     def insert_gridfs(
         self,
         dct: dict,
-        collection: str = "fs",
         compression_type: Optional[Literal["zlib"]] = "zlib",
         oid: Optional[ObjectId] = None,
         task_id: Optional[Union[int, str]] = None,
@@ -268,12 +268,12 @@ def insert_maggma_store(
         task_id: Optional[Any] = None,
     ) -> tuple[int, str]:
         """
-        Insert the given document into a Maggma store, first check if the store is already
+        Insert the given document into a Maggma store.
 
         Args:
             data: the document to be stored
             collection (string): the name prefix for the maggma store
-            oid (str or ObjectId): the _id of the file; if specified, it must not already exist in GridFS
+            oid (str, ObjectId, None): the _id of the file; if specified, it must not already exist in GridFS
             task_id(int or str): the task_id to store into the gridfs metadata
         Returns:
             file id, the type of compression used.
diff --git a/emmet-cli/setup.py b/emmet-cli/setup.py
index 8c29e48d8b..3fbb677e39 100644
--- a/emmet-cli/setup.py
+++ b/emmet-cli/setup.py
@@ -18,7 +18,6 @@
         "click",
         "colorama",
         "mongogrant",
-        "atomate",
         "mgzip",
         "slurmpy",
         "github3.py",