@@ -250,9 +250,12 @@ class ResultContainer(object):
250250
251251 for i, res in enumerate(results):
252252 # FIXME : handle more than one category per engine
253− category = engines[res['engine']].categories[0] + ':' + ''\
254− if 'template' not in res\
255− else res['template']
253+ res['category'] = engines[res['engine']].categories[0]
254+
255+ # FIXME : handle more than one category per engine
256+ category = engines[res['engine']].categories[0]\
257+ + ':' + res.get('template', '')\
258+ + ':' + ('img_src' if 'img_src' in res or 'thumbnail' in res else '')
256259
257260 current = None if category not in categoryPositions\
258261 else categoryPositions[category]