diff --git a/src/planet.py b/src/planet.py
index 3265349..2890360 100755
--- a/src/planet.py
+++ b/src/planet.py
@@ -53,11 +53,11 @@ def insert(self):
     def get(id, name):
         query = {}
         if id is not None:
-            query.update({"_id": id})
+            query["_id"] = id
         if name is not None:
-            query.update({"name": name})
+            query["name"] = name
         try:
-            response = [planet for planet in planets.find(query)]
+            response = list(planets.find(query))
             return {"status": 200, "response": response}
         except:
             return {"status": 500, "response": "Internal server error"}