From 5d4241037408040a83adb573abb7057b69d31972 Mon Sep 17 00:00:00 2001
From: Pawel Srokosz <pawel.srokosz@cert.pl>
Date: Tue, 4 Feb 2020 15:17:25 +0100
Subject: [PATCH] Fixed bug in Malwarecage.search methods

---
 src/core.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/core.py b/src/core.py
index b5d6c02..76b05e8 100644
--- a/src/core.py
+++ b/src/core.py
@@ -416,7 +416,7 @@ def search_files(self, query):
         :rtype: Iterator[:class:`MalwarecageFile`]
         :raises: requests.exceptions.HTTPError
         """
-        return self._recent("file", query)
+        return self._recent(MalwarecageFile, query)
 
     def search_configs(self, query):
         """
@@ -427,7 +427,7 @@ def search_configs(self, query):
         :rtype: Iterator[:class:`MalwarecageConfig`]
         :raises: requests.exceptions.HTTPError
         """
-        return self._recent("config", query)
+        return self._recent(MalwarecageConfig, query)
 
     def search_blobs(self, query):
         """
@@ -438,7 +438,7 @@ def search_blobs(self, query):
         :rtype: Iterator[:class:`MalwarecageBlob`]
         :raises: requests.exceptions.HTTPError
         """
-        return self._recent("blob", query)
+        return self._recent(MalwarecageBlob, query)
 
     @staticmethod
     def _convert_bytes(data):