fix user list

faruken · Aug 25, 2016 · 08463a0 · 08463a0
1 parent 127c431
commit 08463a0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/aiotinder/controllers/tinder.py b/aiotinder/controllers/tinder.py
@@ -2,7 +2,6 @@
 
 """Tinder module.
 """
-
 from typing import (AnyStr, List)
 
 from aiotinder.models.model import User
@@ -35,4 +34,8 @@ async def prospective_matches(self, locale: AnyStr = "en-US") -> List[User]:
         :return: List of users.
         """
         response = await self.tinder.prospective(locale)
-        return [User(**result) for result in response["results"]]
+        res = []
+        for result in response["results"]:
+            if result.get("type") == "user":
+                res.append(User(**result.get("user")))
+        return res